NVIDIA GPU 사용해서 Container 실행.

먼저 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