Ubuntu 16.04 CUDA9.0 Docker 安装 tensorflow-gpu

第一步：准备工作，按照官网https://github.com/NVIDIA/nvidia-docker给出的三个命令安装。

[email protected]_machine:curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
  sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
[email protected]_machine:curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-docker.list
[email protected]_machine:sudo apt-get update

第二步：安装

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

第三步：通过nvidia-smi验证是否成功

docker run --runtime=nvidia --rm nvidia/cuda:9.0-base nvidia-smi

如果出现关于GPU的信息，则表明安装成功。

 

然而，第三步的命令执行后，返回错误信息：

docker: Error response from daemon: Unknown runtime specified nvidia

此时将如下配置内容写到配置文件/etc/docker/daemon.json中即可：

{
    "registry-mirrors":["https://registry.docker-cn.com"],//用于加速下载镜像
    "default_runtime":"nvidia",//用于在输入命令式用docker取代nvidia-docker
    "runtimes":{
        "nvidia":{
            "path":"/usr/bin/nvidia-container-runtime",
            "runtimeArgs":[]
        }
    }
}

然后依次执行以下命令：

sudo systemctl daemon-reload

sudo systemctl restart docker

最后再执行第三步即可。

 

如果闲每次输入docker命令都要加sudo的话，可以执行下面的命令：

sudo groupadd docker     #添加docker用户组
sudo gpasswd -a $USER docker     #将登陆用户加入到docker用户组中
newgrp docker     #更新用户组