Tensorflow配置
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/