pytorch docker + jupyter notebook

1.pytorch docker 파일 다운로드

https://hub.docker.com/r/pytorch/pytorch

https://hub.docker.com/r/pytorch/pytorch

 

2.pytorch image에 jupyter 다운 후 commit

pytorch 이미지파일로 컨테이너 생성 후 컨테이너에 접속하여 pip install jupyter.

그 후 exit로 나온 뒤 commit.

이런식으로 뒤에 -jupyter라고 추가하여 저장하면 편하다.

그 후 기존의 이미지는 삭제해준다.

3.새 이미지 파일로 컨테이너 생성후 접속

docker run -itd --name 원하는이름 -v 마운트할경우입력 -p 8888:8888 --gpus all --restart=always pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime-jupyter

docker exec -it 컨테이너명 bash

4.jupyter 실행

컨테이너 안에서 jupyter notebook --ip=0.0.0.0 --port=8888 --allow-root

이후 나오는 주소를 복사하여 접속하면 pytorch 이미지안에 jupyter를 설치하여 그 안에서 jupyter로 작업이 가능하다.