查看ubantu中自带的python版本,安装对应的tensorflow:
python -V
效果如图,我的是python2.7.12:
准备工作:安装pip
sudo apt-get install python-pip python-dev
根据不同版本的python安装不同的tensorflow(以下爱都是CPU版本的tensorflow,GPU版本可以参见官网):
python2.x
pip install tensorflow
python3.x
pip3 install tensorflow
验证tensorflow是否安装成功:
python
调用成功之后输入以下简单代码进行测试:
import tensorflow as tf
sess = tf.Session()
hello = tf.constant(‘Hello,Tensorflow!’)
print(sess.run(hello))
输入到第二行的时候可能会报错:Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA,如下图:
不是什么大问题,你可以在代码中加上以下两行来忽略这个报错,当然不加直接继续输入也并没有问题,只是看着不太舒服而已:
import os
os.environ[‘TF_CPP_MIN_LOG_LEVEL’] = ‘2’
然后可以看到程序输出**Hello,Tensorflow!**表示安装成功~