本过程是在docker及CUDA安装之后的过程
首先先拉Tensorflow的GPU版本;
docker pull tensorflow/tensorflow:latest-gpu-jupyter
其次;可以使用
docker run -it --name TensorFlow --gpus all -p 9999:8888 -v //path/:/opt --privileged -v /etc/localtime:/etc/localtime:ro tensorflow/tensorflow:latest-gpu-jupyter /bin/bash
但是这样可能会出现如下问题:
docker: Error response from daemon: could not select device driver ““ with capabilities: [[gpu]]问题解决
通过以下命令安装nvidia-container-toolkit
:
sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
然后执行systemctl restart dockerd
或者 systemctl restart docker
重启docker服务便可。