nVidia Docker 설치

1. Docker 설치 (docker-ce)

curl -fsSL https://get.docker.com/ | sudo sh

 

2. nvidia-docker 설치

# 기존 도커(nvidia-docker)와 컨테이너 모두 제거
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 -y nvidia-docker

# 패키지 저장소 추가
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

# nvidia-docker2 설치
Install nvidia-docker2 and reload the Docker daemon configuration
sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd

# 테스트
docker run --runtime=nvidia --rm nvidia/cuda:10.2-base nvidia-smi