nvidia-dockerがver.2になったよ

nvidia-dockerのバージョンが上がった

nvidia-dockerのversion 1.0がdeprecatedになりました。

使い方が以下のように変わります。

nvidia-docker run --rm nvidia/cuda nvidia-smi # これまで
docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi # これから

version 2.0のインストール前には綺麗にversion 1.0を消さないといけません。Installationをしっかり見て更新しましょう。下には参考程度に現段階におけるUbuntu16.04での更新手順を載せます。

Ubuntu16.04における更新手順

旧バージョンの削除

nvidia-docker (ver.1)によって開始された全てのコンテナを停止した後、

ubuntu

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 nvidia-docker

新バージョンのインストール

curl -L https://nvidia.github.io/nvidia-docker/gpgkey | \
sudo apt-key add -
sudo tee /etc/apt/sources.list.d/nvidia-docker.list <<< \
"deb https://nvidia.github.io/libnvidia-container/ubuntu16.04/amd64 /
deb https://nvidia.github.io/nvidia-container-runtime/ubuntu16.04/amd64 /
deb https://nvidia.github.io/nvidia-docker/ubuntu16.04/amd64 /"
sudo apt-get update

sudo apt-get install nvidia-docker2
sudo pkill -SIGHUP dockerd