nvidia-container-toolkitのインストール

Docker 19.03よりNvidia GPUがDocker-runtimeのデバイスとしてサポートされ、
nvidia-docker2がnvidia-container-toolkitに変わった。
すでにnvidia-docker2のパッケージの使用は推奨されていない。

私の場合はNvidia-docker2がインストールしていて
runtimeのオプションが変更されると困る場合があるため、
どちらでも実行できるようにしました。

nvidia-docker2を導入していない場合は以下参照

github.com

nvidia-container-toolkitのインストール

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

sudo apt update && sudo apt install -y nvidia-container-toolkit
sudo systemctl restart docker

nvidia-docker2を使用したアップグレード

sudo apt update
sudo apt --only-upgrade install docker-ce nvidia-docker2
sudo systemctl restart docker

確認

docker run --gpus all nvidia/cuda:9.0-base nvidia-smi
docker run --runtime nvidia nvidia/cuda:9.0-base nvidia-smi
nvidia-docker run nvidia/cuda:9.0-base nvidia-smi

すべてnvidia-smiが表示されればOK

以上。