인프라/Docker
jupyter, tensorflow on docker with gpu (gpu 사용하면서 )
리그캣
2019. 6. 17. 11:12
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
hub.docker.com
$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 에 잡히게 되어 소스코드들이 저장된다.