你不知道的:Intel优化版TensorFlow

你不知道的:Intel优化版TensorFlow,第1张

用常规命令:pip install tensorflow -i http://mirrors.aliyun.com/pypi/simple/

安装TensorFlow后,会出现如下提示:

即官方Build版,没有开启Intel CPU优化,估计是为了兼容AMD的CPU吧

若使用Intel CPU且确认支持AVX AVX2,可以安装Intel优化版的TensorFlow;训练和推理速度可以得到极大提升。安装Intel优化版TensorFlow的步骤:

安装完毕后,运行测试程序

TensorFlow V1.x版本,检测是否使用Intel MKLDNN优化,用命令:

TensorFlow V2.x版本,检测是否使用Intel MKLDNN优化,用命令:

新版tensorflow-gpu, 2.2.0 版推荐的测试程序与之前的1.2.0 不同,记录一下。

(1) jupyter notebook示例

(2)pycharm示例

(1) jupyter notebook

(2) pycharm 测试示例


欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/yw/7973234.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2023-04-12
下一篇 2023-04-12

发表评论

登录后才能评论

评论列表(0条)

保存