1. 查看pip版本

~$pip --version

pip 10.0.1 from /home/xx/anaconda3/lib/python3.6/site-packages/pip (python3.6)

2. 安装tensorflow

~$pip install tensorflow

3. 打开python shell验证是否安装成功

~$python

>>>import tensorflow as tf

>>>hello = tf.constant('Hello, TensorFlow!')

>>>sess = tf.Session()

>>>print(sess.run(hello))

可以成功输出’Hello,TensorFlow!’

但是这时会有警告:

Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2

这是因为tensorflow不支持cpu的AVX2编译,只有CPU版会出现这个问题,GPU不会出现

该问题可以忽略,通过显示等级设置,可设置为不提醒。

>>>>import os  
>>>>os.environ["TF_CPP_MIN_LOG_LEVEL"]='1' # 这是默认的显示等级,显示所有信息  
>>>>os.environ["TF_CPP_MIN_LOG_LEVEL"]='2' # 只显示 warning 和 Error   
>>>>os.environ["TF_CPP_MIN_LOG_LEVEL"]='3' # 只显示 Error

4. 最后

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐