为什么 TensorFlow 会发出关于不支持的 AVX/AVX2 指令的警告，以及如何修复它？

“您的 CPU 支持此 TensorFlow 二进制文件未编译为使用的指令：AVX AVX2”

问题：

我安装了 TensorFlow 并收到一条警告消息，指出我的 CPU 支持二进制文件未编译使用的指令。为什么会发生这种情况，我该如何解决？

答案：

警告是什么？

AVX（高级向量扩展） 和 FMA（融合乘加） 是x86 指令集架构的扩展，提供更快的线性代数计算，如点积、矩阵乘法和卷积。现代 CPU 支持这些扩展，可以显着加快机器学习训练速度。

那为什么不使用它？

默认的 TensorFlow 发行版旨在与尽可能多的 CPU 兼容。它旨在与 GPU 一起使用，在执行大规模训练任务时，GPU 的速度比 CPU 快得多。

要做什么：

如果您有 GPU：

• 使用 os.environ['TF_CPP_MIN_LOG_LEVEL'] 忽略警告= '2'.

如果您没有 GPU 并且想要使用 CPU：

• 从源代码构建 TensorFlow 启用 AVX、AVX2 和 FMA。这需要使用 Bazel 构建系统，如链接讨论中所述。这应该消除警告并提高性能。

以上是为什么 TensorFlow 会发出关于不支持的 AVX/AVX2 指令的警告，以及如何修复它？的详细内容。更多信息请关注PHP中文网其他相关文章！