docker tensorflow - arcturus9/useful-link GitHub Wiki
* Final Setup
https://eungbean.github.io/2018/11/09/Ubuntu-Installation2-3/
* Trial Setup
https://devyurim.github.io/python/tensorflow/development%20enviroment/docker/2018/05/25/tensorflow-3.html
https://github.com/NVIDIA/nvidia-docker/wiki/Installation-(version-1.0)
1. Docker ์ค์น ๋ฐ ์
ํ
(1) docker Install
$ curl -fsSL https://get.docker.com/ | sudo sh
$ sudo usermod -aG docker $USER // ํ์ฌ ์ ์์ค์ธ ์ฌ์ฉ์์๊ฒ ๊ถํ์ฃผ๊ธฐ
$ sudo usermod -aG docker ailab // 'ailab' ์ฌ์ฉ์์๊ฒ ๊ถํ์ฃผ๊ธฐ
$ docker version
2. NVIDIA-docker ์ค์น
(2) NVIDIA-docker
nvidia-docker : https://github.com/NVIDIA/nvidia-docker
nvidia-docker 1.0 ์ค์น ๊ณต์๋ฌธ์ : https://github.com/NVIDIA/nvidia-docker/wiki/Installation-(version-1.0)
$ sudo apt-get install nvidia-docker
$ nvidia-docker run --rm nvidia/cuda nvidia-smi
2. NVIDIA ๋๋ผ์ด๋ธ, CUDA, cuDNN ์ค์น ํ์ธ
NVIDIA ๋๋ผ์ด๋ฒ ์ค์น : https://devyurim.github.io/2018/05/24/15.html
CUDA 9.0, cuDNN 7.0 ์ค์น : https://devyurim.github.io/2018/05/25/16.html
$ cat /usr/local/cuda/version.txt // cuda ์ค์น ํ์ธ
$ cat /usr/include/cudnn.h | grep -E "CUDNN_MAJOR|CUDNN_MINOR|CUDNN_PATCHLEVEL" // cuDNN ์ค์น ํ์ธ
3. Tensorflow Docker ์ค์น
๋์ปค(Docker)๋ ์ด๋ฏธ์ง(image)ํ์ผ์ ์ด์ฉํ์ฌ ๊ฐ๋ฐ ํ๊ฒฝ์ ์ด์ํ ์ ์๋ค. ํ
์ํ๋ก์ฐ์์ ๊ณต์์ ์ผ๋ก ์ ๊ณตํ๋ ํ
์ํ๋ก์ฐ ๋์ปค๋ฅผ ๋ค์ด๋ก๋ ๋ฐ์ ์ ์๋ค.
์ฐ์ ๊ณต์ ๋ฌด๋ฃ ๋ฐฐํฌ ์ด๋ฏธ์ง๋ค์ ๋ค์ด๋ก๋ ๋ฐ์ผ๋ ค๋ฉด docker hub ํํ์ด์ง์ ๊ฐ์
์ด ํ์ํ๋ค.
DockerHub : https://hub.docker.com/
๊ทธ๋ฆฌ๊ณ ํฐ๋ฏธ๋์์ ์๋์ ๋ช
๋ น์ด๋ฅผ ์ด์ฉํ์ฌ docker hub์ ๋ก๊ทธ์ธํด์ค๋ค.
$ docker login
$ docker pull tensorflow/tensorflow:1.8.0-gpu-py3 // ํ๊ทธ๋ช
์ด 1.8.0-gpu-py3์ธ ํ
์ํ๋ก์ฐ ์ด๋ฏธ์ง ๋ค์ด๋ก๋
$ docker images // ๋์ปค ๋ด ์ด๋ฏธ์ง ํ์ธ ๋ช
๋ น์ด
์ด๋ฏธ์ง๋ฅผ ๋ค์ด ๋ก๋ ๋ฐ์์ผ๋ฉด nvidia-docker๋ฅผ ์ด์ฉํ์ฌ ์ปจํ
์ด๋๋ฅผ ์์ฑํ๋ฉด๋๋ค. docker์์ ์ปจํ
์ด๋๋ฅผ ์์ฑํ๋ ๋ช
๋ น์ด๋ run์ด๋ค. ๋ด๊ฐ ์ฌ์ฉํ ์ต์
์ ๋ค์๊ณผ ๊ฐ๋ค.
$ nvidia-docker run -it \
-p 8891:8888 -p 6006:6006
--name ailab-yurim
-v /home/ailab/docker/ailab-yurim:/notebooks \
-e PASSWORD="0000" \
--restart always \
tensorflow/tensorflow:1.8.0-gpu-py3