ubuntu TensorFlow安装

时间:2022-04-24 12:36:27

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