Docker目前也只是跑通,能进行基本的使用。
如果有需求,还可以使用Docker GPU版本,可以使用GPU。
以GPU版本为例:
卸载原有的Docker
sudo apt-get remove docker
sudo apt-get remove docker-ce
sudo apt-get remove docker docker-engine docker.io containerd runc
本机环境
nvidia驱动:最新版
Ubuntu16.04
Python3.6
Tensor2Tensor 1.9 #不要下载最新版,只支持TF1.13.0
Tensorflow 1.12.0 #(可不安装)
Tensorflow-gpu 1.12.0
CUDA 9.0.176
cudnn 7.0.5
pip3 最新版
Tensorflow安装(默认已安装好对应的CUDA和cudnn版本)
sudo pip3 install tensorflow==1.12 #(可不安装)
sudo pip3 install tensorflow-gpu==1.12 #GPU版本
#可以使用清华源进行安装,满速下载
sudo pip3 install tensorflow-gpu==1.12.0 -i https://pypi.tuna.tsinghua.edu.cn/simple #可改成任意版本
Docker CE安装 :官方文档
sudo apt-get update
sudo apt-get