ubuntu TensorFlow安装

Tensorflow安装确实比caffe简单。。。

cuda，cudnn安装就不说了

1 安裝pip（有就跳过）

sudo apt-get install python-pip python-dev   # for Python 2.7
sudo apt-get install python3-pip python3-dev # for Python 3.n

2 安装tensorflow-gpu版本

pip install tensorflow-gpu  # Python 2.7;  GPU support
pip3 install tensorflow-gpu # Python 3.n;  GPU support

这样默认安装的是最新版本(目前是1.4.0)，但我用的是cudnn 5，最新的不支持，因此安装其他特定版本，比如我安装1.1.0

sudo pip install --upgrade https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.1.0-cp27-none-linux_x86_64.whl   # CPU

sudo pip install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.1.0-cp27-none-linux_x86_64.whl  # GPU

 如果下载很慢，就试试下面的方法：

sudo apt-get install libcupti-dev
sudo apt-get install python-pip python-dev python-virtualenv
virtualenv --system-site-packages ~/tensorflow-gpu
# 开 VPN 下载 whl 格式的安装包，速度快。
wget https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.1.0-cp27-none-linux_x86_64.whl
source ~/tensorflow-gpu/bin/activate
# 从豆瓣的镜像安装其它依赖包
(tensorflow-gpu)$ pip install tensorflow_gpu-1.1.0-cp27-none-linux_x86_64.whl -i http://pypi.douban.com/simple/ --trusted-host=pypi.douban.com

 

查看tensorflow 的版本

python -c 'import tensorflow as tf; print(tf.__version__)'# for Python 2

python3 -c 'import tensorflow as tf; print(tf.__version__)'# for Python 3

测试是否安装成功：

>>> import tensorflow as tf  
>>> hello = tf.constant('Hello, TensorFlow!')  
>>> sess = tf.Session()  
>>> print(sess.run(hello))