ubuntu系統nvidia-docker安裝指導

為什麼要安裝nvidia-docker?

•  docker映象中可以安裝CUDA、cuDNN，但是映象中沒有顯示卡驅動，因此如果要在映象中使用GPU，那麼需要將本機的顯示卡驅動對映到Docker映象中，這就需要安裝nvidia-docker了，nvidia-docker的任務就是將顯示卡驅動對映到Docker映象中。

nvidia-docker主要版本

• nvidia-docker1
• nvidia-docker2

nvidia-docker2安裝方法

• 如果已經安裝了nvidia-docker1，可以通過以下命令解除安裝和移除：
  • docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -nl docker ps -q -a -f volume={} | xargs -r docker rm -f 
  • sudo apt-get purge -y nvidia-docker
  • 接下來新增GPG key並建立一個倉庫
  • 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
  • sudo apt-get update
  • 開始安裝nvidia-docker2：sudo apt-get install -y nvidia-docker2
  • 重新匯入Docker配置檔案：sudo pkill -SIGHUP dockerd
  • 測試nvidia-docker2是否安裝成功：docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi（其中nvidia/cuda是映象名）

去除--runtime=nvidia方法

• vi /etc/docker/daemon.json
• 改成如下內容

{
    "default-runtime": "nvidia",
    "runtimes": {
        "nvidia": {
            "path": "nvidia-container-runtime",
            "runtimeArgs": []
        }
    }
}