经过Intel® MKL-DNN优化后,TensorFlow在CPU上的训练速度也可以得到提升,参考《现代英特尔® 架构上的 TensorFlow* 优化》
Intel自己给出了很多现成编译好的,经过优化后的TensorFlow安装包,参考《英特尔® Optimization for TensorFlow* 安装指南》
Anaconda社区也为此做出了巨大贡献,在Windows上,一条指令:
conda install tensorflow-mkl -c anaconda
就可以完成Intel优化版的TensorFlow安装。
对于同一个CNN训练,经过编译优化的版本,训练速度明显提升,如下图所示:17s vs 24s, 30%提升。
网友评论