docker启动容器时运用宿主机的GPU

在装好docker以及nvidia-docker后，启动Ubuntu18/cuda的docker镜像
先进行一个测试，执行以下命令

$ docker run --runtime=nvidia --rm xxx nvidia-smi

注意：xxx为你的镜像名称
如果有类似以下输出，则表示docker可以正常调用GPU

然后执行如下命令即可进入容器终端

$ docker run --runtime=nvidia -it xxx /bin/bash

如果报以下错

则到/etc/docker/文件夹下新建daemon.json文件（如果文件存在就不用新建了），并写入以下内容：

{
    "runtimes": {
        "nvidia": {
            "path": "/usr/bin/nvidia-container-runtime",
            "runtimeArgs": []
        }
    }
}

一定要注意json文件的格式要正确。

然后重启docker服务：

$ systemctl restart docker
$ systemctl daemon-reload

Done!