Pycharm下安装 Tensorflow

 

 

1. 使用 pip 下载TensorFlow

  Pycharm命令提示行中输入：

pip install tensorflow==2.0.0a0

2.测试是否配置成功

在新建的项目中创建.py文件，运行

import tensorflow as tf

if __name__ == '__main__':
    # 定义两个向量a,b
    a = tf.constant([1.0, 2.0], name='a')
    b = tf.constant([2.0, 3.0], name='b')
    result = a + b
    sess = tf.Session()  # 生成一个会话，通过一个会话session来计算结果
    # 实现了一个简单的Tensorflow模型
    print(sess.run(result))

遇到错误：

原因是下载TensorFlow的版本不支持cpu的AVX2编译。

可能是因为安装时使用的pip install tensorflow ，这样默认会下载X86_64的SIMD版本。

有两种解决办法：

1.忽略这个警告

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

2.换成支持cpu用AVX2编译的TensorFlow版本。

首先，卸载原来版本的TensorFlow 

pip uninstall tensorflow

完成后可以用pip list查看一下所有库，检查是否成功卸载。

然后去github下载正确的tf版本，Windows点这里下载。其他操作系统可以点这里找到对应的版本。

对应的path在上边的找到对应的.whl下载即可。

然后安装：

pip install tensorflow-1.11.0-cp37-cp37m-win_amd64.whl

运行成功：