TensorFlow二进制优化为在性能关键操作中使用以下CPU指令:AVX2 FMA

2024-10-06 12:22:44 发布

您现在位置:Python中文网/ 问答频道 /正文

此TensorFlow二进制文件通过oneAPI深层神经网络库(oneDNN)进行优化,以便在性能关键型操作中使用以下CPU指令:AVX2 FMA要在其他操作中启用它们,请使用适当的编译器标志重建TensorFlow

试图运行一个使用Tensorflow库的程序,我在PyCharm IDE for macOS上遇到了这个错误。为什么仅仅安装Tensorflow还不够好?我按照此处的安装说明操作:https://www.tensorflow.org/install/pip


Tags: 文件编译器tensorflow指令二进制神经网络cpu性能
1条回答
网友
1楼 · 发布于 2024-10-06 12:22:44

我通过如下设置环境变量修复了此错误

os.environ['KMP_DUPLICATE_LIB_OK'] = 'TRUE'

它在Ubuntu16.04和Python3.7(Anaconda)上完美地工作

相关问题 更多 >