安装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 测试示例
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)