windows版TensorFlow最优安装,使用AVX2指令集

windows版TensorFlow最优安装,使用AVX2指令集通常我们运行TensorFlow会报告如下信息,意思是你的CPU支持AVX2指令集,但TensorFlow的二进制版本没有使用2019-02-1415:44:41.989265:IT:\src\github\tensorflow\tensorflow\core\platform\cpu_feature_guard.cc:141]YourCPUsupportsinstruction…

大家好,又见面了,我是你们的朋友全栈君。

通常我们运行TensorFlow会报告如下信息,意思是你的CPU支持AVX2指令集,但TensorFlow的二进制版本没有使用

2019-02-14 15:44:41.989265: I T:\src\github\tensorflow\tensorflow\core\platform\cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2

对于有强迫症的我来说难以容忍,由于GPU GTX1050(2G)内存不够,不能用于训练,全靠CPU来训练,CPU训练是相当花时间的,于是琢磨使用AVX2指令集

要使用AVX2指令集,正规的做法是自己编译TensorFlow,刚开始我也是这么干的,各种软件下载,环境配置,各种问题解决,然后编译,这个非常耗时,在我的笔记本上大概要3-4小时,忙活完了一天都快过去了,安装后发现我的编译参数设置不对,还要重新编译,崩溃

幸好发现一个好东西,省去了这些麻烦,这个网站编译好了各种版本的TensorFlow,下载就可以使用

https://github.com/fo40225/tensorflow-windows-wheel

我使用的python 3.6,需要下载下面的版本,注意要使用avx2版本

tensorflow_gpu-1.12.0-cp36-cp36m-win_amd64.whl (cuda100cudnn73avx)

tensorflow-1.12.0-cp36-cp36m-win_amd64.whl (avx2)

cuda需要使用10.0.X

cudnn需要使用7.3.X

(X尽量选大的版本,具体安装过程就不详细说了)

可以说这种安装办法是目前最好,最省时间的办法,充分利用cpu的指令集,避免的cuda和cudnn版本的烦恼。

用pip网络安装TensorFlow,一是没有cpu优化,二是cuda和cudnn版本不清楚,会出现大量问题花大量时间

  • 问题

1.msgpack版本问题

安装Keras的过程中,遇到下面错误,找不到1.21.8版的msgpack

distributed 1.21.8 requires msgpack, which is not installed.

解决办法是安装最新版

pip install msgpack

2.h5py的版本问题

执行keras-yolo的时候出现下面的警告

C:\Users\xxx\AppData\Local\Continuum\anaconda3\lib\site-packages\h5py\__init__.py:36: FutureWarning: Conversion of the second argument of issubdtype from `float` to `np.floating` is deprecated. In future, it will be treated as `np.float64 == np.dtype(float).type`.
  from ._conv import register_converters as _register_converters

解决办法是安装最新版

pip install –upgrade h5py

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请联系我们举报,一经查实,本站将立刻删除。

发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/142874.html原文链接:https://javaforall.net

(0)
全栈程序员-站长的头像全栈程序员-站长


相关推荐

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

关注全栈程序员社区公众号