[Linux] nvidia-docker 2 설치하기

1. 준비 사항

1. GNU/Linux x86_64 with kernel version > 3.10
2. Docker >= 1.12
3. NVIDIA GPU with Architecture > Fermi (2.1)
4. NVIDIA drivers ~= 361.93 (untested on older versions)

드라이버 버전에 따라 사용할 수 있는 CUDA 버전이 있으니 이를 참고하세요.

1.1 레포지토리 구성하기

Debian-based 배포판에서는:

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
  sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update

RHEL-based 배포판에서는:

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | \
  sudo tee /etc/yum.repos.d/nvidia-docker.repo

1.2 레포지토리 키 업데이트하기

Debian-based 배포판에서는:

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
  sudo apt-key add -

RHEL-based 배포판에서는:

DIST=$(sed -n 's/releasever=//p' /etc/yum.conf)
DIST=${DIST:-$(. /etc/os-release; echo $VERSION_ID)}
sudo rpm -e gpg-pubkey-f796ecb0
sudo gpg --homedir /var/lib/yum/repos/$(uname -m)/$DIST/nvidia-docker/gpgdir --delete-key f796ecb0
sudo yum makecache

2. nvidia-docker 1.0 버전 삭제하기

nvidia-docker 1.0으로 실행된 모든 컨테이너를 중지하고 삭제합니다.

2.1 Ubuntu 배포판

docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f
sudo apt-get purge nvidia-docker

2.2 CentOS 배포판

docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f
sudo yum remove nvidia-docker

3. nvidia-docker 2.0 버전 설치하기

설치 전에 NVIDIA 드라이버와 사용하시는 배포판에 지원되는 Docker 버전이 설치되었는지 확인하세요!

3.1 Ubuntu 배포판

sudo apt-get install nvidia-docker2
sudo pkill -SIGHUP dockerd

3.2 CentOS 배포판

sudo yum install nvidia-docker2
sudo pkill -SIGHUP dockerd

4. 기본 사용방법

nvidia-docker는 새 컨테이너 런타임을 Docker 데몬에 등록합니다.

docker run 을 입력해서 nvidia 런타임을 선택할 수 있습니다:

docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi

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