NVIDIA GPU로 Docker 컨테이너 실행하기

NVIDIA GPU로 Docker 컨테이너 실행하기

먼저 docker를 설치 한다. 블로그에 도커 설치 관련 블로그가 있다. 참고하면 된다. 도커가 설치가 되면. nvidia-gpu를 사용하기 위해서 nvidia 드라이버를 서버에 설치 해야 한다.

nvidia 드라이버 설치 방법

## NVIDIA 드라이버의 최신 버전을 얻기 위해 NVIDIA PPA를 추가.
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update

## 설치 가능한 최신 드라이버를 확인하고 설치.
ubuntu-drivers devices
## 특정 버전 설치.
sudo apt-get install nvidia-driver-XXX
## 드라이버 설치가 완료되면, 시스템을 재부팅하여 드라이버를 활성화.
sudo reboot
## 재부팅 후, 드라이버가 제대로 설치되었는지 확인하려면 다음 명령어.
nvidia-smi

버전 선택 시 참고

서버에 nvidia-smi가 잘 떴다면,

이제, docker-container에서 GPU를 사용할 수 있도록 nvidia-docker2를 설치해야 한다.

## 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
## nvidia-docker 설치.
$ sudo apt-get update
$ sudo apt-get install -y nvidia-docker2
## 컨테이너 생성 후 확인해보기.
docker run --rm --gpus all ubuntu:18.04 nvidia-smi