[딥러닝 서버 A to Z] 도커(Docker) 컨테이너 설정 - 2. NVIDIA-Docker

 

Container에서 GPU를 사용하기 위해서는 NVIDIA Docker를 설치해야 합니다.

 

1. NVIDIA-Docker GPG 및 저장소 추가

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

 

2. NVIDIA-Docker 설치

sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit

 

3. Docker 재실행

sudo systemctl restart docker