确定要安装哪种 TensorFlow
- 仅支持 CPU 的 TensorFlow。如果您的系统没有 NVIDIA® GPU,就必须安装此版本。请注意,此版本的 TensorFlow 通常更容易安装(用时通常在 5 或 10 分钟内),所以即使您拥有 NVIDIA GPU,我们也建议先安装此版本。
- 支持 GPU 的 TensorFlow。TensorFlow 程序在 GPU 上的运行速度通常要比在 CPU 上快得多。因此,如果您的系统配有满足以下所示先决条件的 NVIDIA® GPU,并且您需要运行性能至关重要的应用,则最终应安装此版本。
使用 Anaconda 进行安装
从Anaconda官网下载
ubuntu@VM-54-14-ubuntu:~/downloads$ sudo bash Anaconda3-5.1.0-Linux-x86_64.sh
选择yes
添加环境变量:ubuntu@VM-54-14-ubuntu:~/downloads$ sudo vi ~/.bashrc
在最后一行添加export PATH=/home/ubuntu/anaconda3/bin:$PATH
更新环境变量
ubuntu@VM-54-14-ubuntu:~/downloads$ source ~/.bashrc
创建虚拟环境
conda create -n tensorflow pip python=2.7 # or python=3.3
激活 conda 环境
ubuntu@VM-54-14-ubuntu:~/downloads$ source activate tensorflow_env
查看虚拟环境
ubuntu@VM-54-14-ubuntu:~/downloads$ conda env list
关闭虚拟环境
source deactivate
安装TensorFlow
(tensorflow_env) ubuntu@VM-54-14-ubuntu:~/downloads$ pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.6.0-cp36-cp36m-linux_x86_64.whl