Etc/Docker
[Docker] PyTorch 환경 구축
_JSP_
2023. 1. 16. 22:33
< 기본 설정 >
더보기
1. Docker 설치 및 NVIDIA 관련 설치가 완료되었다고 가정한다.
2. 환경 구축은 Linux 서버 상에서 Docker을 사용하여 진행된다.
3. GPU을 사용하는 환경이라고 가정한다.
1. Docker 이미지 다운로드
docker pull pytorch/pytorch
2. PyTorch 컨테이너 생성
docker run -itd --name pytorch_study --gpus '"device=1"' -p 8888:8888 -p 8889:8889 pytorch/pytorch- 8888 및 8889 2개의 포트를 열었다.
- GPU가 여러 개이기 때문에 1개의 GPU만을 지정하였다.
3. PyTorch 컨테이너 접속
docker exec -it [컨테이너ID] bash
4. 필요 패키지 설치
설치는 Conda install로 진행한다.
(패키지 충돌을 방지하기 위함)
1. Jupyter notebook
conda install jupyter
5. Jupyter Notebook 실행(백그라운드)
nohup jupyter notebook --ip=0.0.0.0 --port=8888 --allow-root --NotebookApp.token='' &