如何利用具有张量流的intel-mkl

Question

我已经看过很多关于使用具有张量流的CPU的文档,但是,我没有GPU.我所拥有的是一个相当强大的CPU和一个5GB的英特尔数学内核,我希望这可以帮助我加快张力流的速度.

有谁知道我怎么能"制造"tensorflow使用intel-mlk？

Answer 1

由于tensorflow使用Eigen，请尝试使用启用MKL的Eigen版本，如下所述：

1. 在包含任何 Eigen 标头之前定义 EIGEN_USE_MKL_ALL 宏
2. 将您的程序链接到 MKL 库（请参阅MKL 链接顾问）
3. 在 64 位系统上，您必须使用 LP64 接口（而不是 ILP64 接口）

所以一种方法是按照上面的步骤修改tensorflow的源，重新编译并安装在你的机器上。当您使用它时，您还应该尝试英特尔编译器，如果您设置正确的标志，它本身可能会提供不错的性能提升：-O3 -xHost -ipo。