一、安装Docker
1、依次执行下面命令进行安装。
$ sudo apt install docker.io
$ sudo systemctl start docker
2、检查是否安装成功:
$ docker -v
Docker version 17.12.1-ce, build 7390fc6
二、安装TensorFlow
1、下载tensorflow镜像:
$ docker run -it tensorflow/tensorflow:1.4.0
2、安装最新的Nvidia驱动以及nvidia-docker:
https://blog.youkuaiyun.com/menghuanbeike/article/details/79212195
3、通过pip 安装 TensorFlow
- 安装pip:
在Ubuntu/linux-64-bit环境下安装:
sudo apt-get install python-pip python-dev
- 找到合适的安装包URL
支持CPU的TensorFlow安装包-python3.5环境:
export TF_BINARY URL-https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.9.0-cp35-cp35m-linux x86 64.whl
支持CPU的TensorFlow pip安装包的url-python3.5环境:
export TE_BINARY URL-https://storage.googleapis.com/tensorflow/linux/
gpu/tensorflow-0.9.0-cp35-cp35m-linux x86 64.whl
通过pip安装TensorFlow
sudo pip3 install –upgrade $TE BINARY URL。
通过这三步,TensorFlow环境就安装完成了。
参考《实战Google深度学习框架》书籍。