用于 mac AVX 的 TensorFlow

Posted

技术标签:

【中文标题】用于 mac AVX 的 TensorFlow【英文标题】:Tensorflow for mac AVX 【发布时间】:2018-03-19 17:03:37 【问题描述】:

我正在使用 TensorFlow,但收到以下警告

TensorFlow 库未编译为使用 AVX 指令,但 这些在您的机器上可用,并且可以加速 CPU 计算

我是用 pip 安装的。

我该如何解决?

【问题讨论】:

@sophros - 请不要使用 code formatting 来突出显示,或者代码中不是特别变量或函数名称的名称。它看起来很丑陋,而且显然给屏幕阅读器带来了麻烦。 【参考方案1】:

这是因为你使用 pip 安装的,它是一个预编译包,没有预编译到你使用的架构,你需要从源代码构建它。

你可以参考他官方的tensorflow文档了解如何从源码编译

https://www.tensorflow.org/install/install_sources

【讨论】:

以上是关于用于 mac AVX 的 TensorFlow的主要内容,如果未能解决你的问题,请参考以下文章

用于复数乘法的汇编代码/AVX 指令。 (GCC 内联汇编)

用于灰度到 ARGB 转换的 C++ SSE2 或 AVX2 内在函数

我在哪里可以找到 AVX 指数双精度函数?

用于整个 256 位寄存器的 AVX unpackhipd/unpacklopd 模拟

AVX/AVX2 是不是“存在”在每个内核上?

SSE/AVX 向量类型的差异