nvidia-docker

Tweet

nvidiaのランタイムをインストールする

nvidiaのイメージをdockerで実行するには、事前にnvidiaのランタイムをインストールしておく必要がある。

sudo apt-get install nvidia-container-runtime

インストールして実行したときに、以下のエラーが出たときは、/etc/docker/daemon.jsonを編集する。

docker: Error response from daemon: Unknown runtime specified nvidia.
See 'docker run --help'.

sudo tee /etc/docker/daemon.json <<EOF
{
    "runtimes": {
        "nvidia": {
            "path": "/usr/bin/nvidia-container-runtime",
            "runtimeArgs": []
        }
    }
}
EOF

teeコマンドは、「コマンドの結果をファイルに出力したいけど、標準出力でも出力を見たい！」場合に使用するコマンド。

イメージのダウンロードと実行

docker run --runtime=nvidia --name darknet -it nvidia/cuda:9.1-devel-ubuntu16.04 /bin/bash

とか

docker run  --runtime=nvidia --name darknet -it nvidia/cuda:9.1-cudnn7-devel /bin/bash