[Docker] PyTorch 환경 구축

< 기본 설정 >

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='' &