nvidia-docker 대신 docker run --gpus로

GPU를 활용가능한 도커 컨테이너들을 관리하기 위해 nvidia가 만든 nvidia-docker가 

명령어가 기존의 docker 커맨드에 통합되는 듯 하다.

 

기존에 cpu 버전은 docker 명령어로 gpu버전은 nvidia-docker 명령어를 사용했는데 

최근 (2019년 11월 현재 기준) 다시 서버를 세팅하면서 보니 docker 버전에 따라 다르지만 점차 docker 

명령어의 태그 표현으로 사용할 수 있게 바뀌었다. 

 

docker를 설치하고 나서 nvidia-docker를 설치하고 나서 보니 nvidia-docker 명령어가 아예 없어서 처음엔 설치가 잘못된 것을 의심했는데 그게 아니었다.

 

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

NVIDIA/nvidia-docker

 

깃헙의 예제를 보면 다음과 같다. 

$ docker run --gpus 2 nvidia/cuda:9.0-base nvidia-smi

nvidia-docker 명령 대신 docker 명령을 사용해도 되며 --gpus 태그로 컨테이너를 할당할 gpu 갯수나 번호등을 할당해 사용할 수 있다.  issue 보드에도 왜 nvidia-docker 명령이 설치했는데도 안되느냐 말들이 보이는 것은 명령어 자체가 바뀌어 생기는 문제인것 같다. 물론 nvidia-docker 명령을 사용하기 위한 방법이 깃헙 페이지에 소개는 되고 있으나 미래에 지원 자체가 끊길 것임을 암시하고 있다.