docker를 활용한 딥러닝 개발환경 구축

Docker를 활용하면, 여러가지 환경을 사용할 수 있다는 장점은 분명 존재합니다.

하지만, docker의 container 내에서 GPU를 사용하는 것은 쉽지는 않습니다.

딥러닝 연구&개발이 급진적으로 진행되고 있는 추세에 대해서, nvidia에서는 docker에서 쉽게 GPU를 사용할 수 있게 오픈 소스를 공개하였습니다.

그 이름은 바로 nvidia-docker 입니다.

아래에 참고 1에 링크를 따라가면, 설치방법이 나와있습니다.

docker에서 image를 다운로드 할 때, lateset 버전을 받으면 CUDA 9.0 버전이 필요하게 됩니다. 

현재 서버에 9.0이 아닌 다른 버전으로 세팅을 해놓으셨다면, 이전 버전에 대해서도 TAG별로 구분하여 다운로드 받을 수 있게 되어있으니 참고하세요.

TAG 명은 아래 참고2에서 확인하실 수 있습니다.

참고; 

1. https://github.com/NVIDIA/nvidia-docker

2. https://hub.docker.com/r/nvidia/cuda/