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 .n
2 安装tensorflow-gpu版本
pip install tensorflow-gpu # Python 2.7; GPU support
pip3 install tensorflow-gpu # Python .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.-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 python3 -c 'import tensorflow as tf; print(tf.__version__)'# for Python
测试是否安装成功:
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))