Tensorflow配置

2023年10月22日 191次阅读 来源: 牧 天

1、安装python 3.6 (目前tensorflow不支持python 3.7)
python版本说明：
Windows x86-64 embeddable zip file 这个属于嵌入式版本
Windows x86-64 executable installer 这个是可执行安装版本
Windows x86-64 web-based installer 这个是有网络才能安装的版本
这里我们选择Windows x86-64 executable installer
安装的时候最好选择添加环境变量当当前用户

2.安装sensorflow
pip install sensorflow
使用这种安装方式，可能会遇到下面这个错误（也有可能不会遇到）
I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 Device mapping: no known devices.
原因是下载TensorFlow的版本不支持cpu的AVX2编译。
可能是因为安装时使用的pip install tensorflow ，这样默认会下载X86_64的SIMD版本。

2种解决办法
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版本
先卸载原来安装的sensorflow
pip uninstall sensorflow
然后去github下载正确的tf版本
https://github.com/fo40225/tensorflow-windows-wheel/blob/master/1.9.0/py36/CPU/avx2/tensorflow-1.9.0-cp36-cp36m-win_amd64.whl
然后安装：
pip install tensorflow-1.6.0-cp36-cp36m-win_amd64.whl
第一种办法未做尝试，这里选用第二种办法

3. 测试程序

  //test.py

import tensorflow as tf
a = tf.constant([1.0, 2.0, 3.0], shape=[3], name='a')
b = tf.constant([1.0, 2.0, 3.0], shape=[3], name='b')
c = a + b
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
print(sess.run(c))

D:\python>python tt.py
Device mapping: no known devices.
2018-12-06 11:31:18.827630: I c:\users\user\source\repos\tensorflow\tensorflow\core\common_runtime\direct_session.cc:288] Device mapping:

add: (Add): /job:localhost/replica:0/task:0/device:CPU:0
2018-12-06 11:31:18.829630: I c:\users\user\source\repos\tensorflow\tensorflow\core\common_runtime\placer.cc:886] add: (Add)/job:localhost/replica:0/task:0/device:CPU:0
a: (Const): /job:localhost/replica:0/task:0/device:CPU:0
2018-12-06 11:31:18.831630: I c:\users\user\source\repos\tensorflow\tensorflow\core\common_runtime\placer.cc:886] a: (Const)/job:localhost/replica:0/task:0/device:CPU:0
b: (Const): /job:localhost/replica:0/task:0/device:CPU:0
2018-12-06 11:31:18.832630: I c:\users\user\source\repos\tensorflow\tensorflow\core\common_runtime\placer.cc:886] b: (Const)/job:localhost/replica:0/task:0/device:CPU:0
[2. 4. 6.]

 

 

TF官网: http://www.tensorfly.cn/

    原文作者：牧 天
    原文地址: https://www.cnblogs.com/muhe221/articles/10075837.html
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞