根据自己的情况选择以下命令之一进行安装:
pip install tensorflow==1.2 # Python 2.7; 仅支持CPU
pip3 install tensorflow==1.2 # Python 3.n; 仅支持CPU
pip install tensorflow-gpu==1.2 # Python 2.7; 支持CPU
pip3 install tensorflow-gpu==1.2 # Python 3.n; 支持CPU
注意这里的==1.2.是表示tensorflow的版本,请根据实际情况修改。cuda8.0对应的是1.2 。
cuda与tensorflow的版本一定要对应,不然会出错!-------重要
如下载速度慢或者老是出现下载中断,解决办法:https://blog.csdn.net/luhuaxiang/article/details/90754795
下图是对照表:
验证Tensorflow是否安装成功
启动终端,输入python,如下:
python
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))
如果输出Hello, TensorFlow!则代表安装成功,如下图。
查看Tensorflow版本
sudo pip show tensorflow
Tensorflow卸载:
sudo pip uninstall protobuf
sudo pip uninstall tensorflow