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))