[Docker] Pytorch쓰기..

서버에 도커는 깔려 있으니까 일단 pytorch docker 이미지 다운받음 서버 쿠다 버전에 맞게 다운받아준다. 

docker pull anibali/pytorch:1.7.0-cuda11.0

 

다음으로 컨테이너 만들고 bash 실행 

docker run -i -t --name juhee-pytorch anibali/pytorch /bin/bash

 

됐당 ㅎ 

컨테이너 나갔다가 다시 실행하고 싶을 때 

docker attach juhee-pytorch

 

문제 생김.. cuda 가 안 잡힘 ^^ ;; nvidia-docker 로 설치해주자 

아까 잘못 깔았던 이미지 삭제 

docker rmi [이미지 이름]

 

가장 최근 버전 이미지 다운받기 

 nvidia-docker pull pytorch/pytorch

 

다시 컨테이너 만들어서 실행 ^_^ 

nvidia-docker container run -it --name "juhee-pytorch" pytorch/pytorch:latest

됐당  gpu 잡힌다 

이제 다시.. 포트 열어서 컨테이너 만들기 

nvidia-docker run -it -p 8888:8888 pytorch/pytorch:latest

 

그리고 주피터 노트북 깔고 실행 

jupyter notebook --ip=0.0.0.0 --allow-root

 

이제 

http://서버ip:port 

로 접속하면 주피터 노트북 들어가진다.. https 가 아니라 http로 해야 접속이 가능했는데 (나 같은 경우에는 ㅎㅎ ) 참고바람.. 

GPU 잘 잡힌다 굳굳 ~~ >_<