ubuntu 16.04 使得自己定制的镜像启动的容器可以使用cuda

前提：你的主系统已经安装了cuda、cudnn、docker，并且可以使用。

解决问题：使得用docker build命令构建的镜像启动的容器中可以使用cuda，继而可以使用tensorflow-gpu。

解决方法：

第一步：Docker Engine setup，参考自https://github.com/nvidia/nvidia-container-runtime#docker-engine-setup

Systemd drop-in file，执行以下命令：

sudo mkdir -p /etc/systemd/system/docker.service.d
sudo tee /etc/systemd/system/docker.service.d/override.conf <<EOF
[Service]
ExecStart=
ExecStart=/usr/bin/dockerd --host=fd:// --add-runtime=nvidia=/usr/bin/nvidia-container-runtime
EOF
sudo systemctl daemon-reload
sudo systemctl restart docker

Daemon configuration file，执行以下命令：

sudo tee /etc/docker/daemon.json <<EOF
{
    "runt