Linux Mint に nvidia-dockerをインストール

Ubuntuとの対応の確認

WikipediaでLinux Mintのバージョンと対応するUbuntuのディストリビューションを確認。

dockerのインストール又はバージョンアップ

以下の手順に従ってdockerをインストール又はバージョンアップする。

ただし、公式サイトの手順でapt-get-repositoryをやると$(lsb_release -cs)でディストリビューション名がUbuntuと違ってコケる。

Ubuntu用スクリプト

$ sudo add-apt-repository \
   "deb [arch=amd64] https://download.docker.com/linux/ubuntu \
   $(lsb_release -cs) \
   stable"

そこで、「Ubuntuとの対応の確認」で確認したディストリビューション名を$(lsb_release -cs)の代わりにベタ書きする。筆者の環境はLinux Mint sarahであったので、対応するディストリビューションはxenialである。

LinuxMint用スクリプト

$ sudo add-apt-repository \
   "deb [arch=amd64] https://download.docker.com/linux/ubuntu \
   xenial \
   stable"

nvidia-dockerのインストール

以下の手順にしたがってインストールする。

ただし、またしてもリポジトリの登録のときのディストリビューション確認でコケる。$distributionに適切な文字列を入れてやらないといけない。

Ubuntu用スクリプト

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
  sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
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ではubuntu18.04、ubuntu16.04、ubuntu14.04の3種類が選択できるようだ。筆者の環境はLinux Mint sarahであったので、対応するディストリビューションはubuntu16.04が対応するので、以下のようにかえてやる。

LinuxMint用スクリプト

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
  sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu16.04/nvidia-docker.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-docker.list

動作確認

手順にしたがって動作確認したところちゃんとグラボを認識してくれた。

確認結果

$ docker run --runtime=nvidia --rm nvidia/cuda:10.2-base nvidia-smi
Unable to find image 'nvidia/cuda:latest' locally
latest: Pulling from nvidia/cuda
（略）
Status: Downloaded newer image for nvidia/cuda:latest
Sun Aug 26 14:52:25 2018       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 384.130                Driver Version: 384.130                   |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  Quadro K620         Off  | 00000000:01:00.0  On |                  N/A |
| 35%   49C    P8     1W /  30W |    336MiB /  1997MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+

結論

最初からubuntuにしとけばよかった・・・・。