docker安装tensorflow-gpu，亲测有用

https://blog.csdn.net/xiangxianghehe/article/details/81216424

docker安装参考以上

请记住到第三步，如果docker安装成功，就测试下。然后下面久不用安装nvidia-docker2了。参考下面链接就可以了。

https://zhuanlan.zhihu.com/p/64416453

tensorflow-gpu安装参考以上

最后结果就是装上了。

我之前用docker pull tensorflow不能满足gpu的需求，自己搞过

docker pull nvidia/cuda也失败了。

知乎上的这个例子是能把两者结合起来，不知道具体原理，亲测有用。最后在jupyter中生成一个连接。

你只要把服务器的地址换上去就可以了。

 

 apt-get remove docker docker-engine docker.io  #全新安装时，无需执行该步骤

sudo apt-get update 

sudo apt-get install \
    apt-transport-https \
    ca-certificates \
    curl \
    software-properties-common

curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -

sudo add-apt-repository \
   "deb [arch=amd64] https://download.docker.com/linux/ubuntu \
   $(lsb_release -cs) \
   stable"

sudo apt-get update

sudo apt-get install docker-ce

此时可以
sudo docker pull hello-world
测试。
接下来是安装nvidia-docker2：



docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f
sudo apt-get purge -y nvidia-docker

# Add the package repositories
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

# Install nvidia-docker2 and reload the Docker daemon configuration
sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd

# Test nvidia-smi with the latest official CUDA image
sudo docker run --runtime=nvidia --rm nvidia/cuda  nvidia-smi
如果现实出来nvidia-smi，表示安装成功。
接下来
sudo docker run -it --net=host--runtime=nvidia -p 8866:8888 tensorflow/tensorflow:latest-gpu-py3-jupyter
记住--net=host，是一定要加的，否则你在docker里面无法连接github。
这个命令如果完成了，会生成一个jupyter notebook，生成的链接你可以在浏览器中输入服务器地址和端口：8888。输入连接后面的token，ok了，
最终，需要进入jupyter notebook里面开启一个终端，输入
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
如果显示有GPU，
那么恭喜你，这个tensorflow可以调用GPU工作。