前提:你的主系统已经安装了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