jupyter, tensorflow on docker with gpu (gpu 사용하면서 )

GPU 2번을 사용하는 Jupyter notebook on docker를 구축해야 하는데 아래 링크가 많은 도움이 되었다

 

http://moducon.kr/2018/wp-content/uploads/sites/2/2018/12/leesangsoo_slide.pdf

불러오는 중입니다...

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

Docker Hub

$docker run --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES={GPU번호} -it -p {호스트PORT}:8888 -v {HOSTDIR}:/tf/notebooks tensorflow/tensorflow:latest-py3-jupyter 

 

 

실제로 나의 명령어는 다음과 같다./

$docker run --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=2 -it -p 4658:8888 -v /home/youngjune/notebooks:/tf/notebooks tensorflow/tensorflow:latest-py3-jupyter

 

위와같이 돌리게되면

http://localhost:4658로 들어가면되고 volume은 /home/leaguecat 에 잡히게 되어 소스코드들이 저장된다.