TensorFlow · 2024年10月31日

TensorFlow AMD运行报错

TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.

TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.
To enable the following instructions: SSE SSE2 SSE3 SSE4.1 SSE4.2 AVX2 AVX512F AVX512_VNNI FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.
JavaScript

这个警告是 TensorFlow 在启动时输出的信息,提醒用户一些当前 TensorFlow 二进制文件已经经过优化,以利用充分可用的 CPU 指令集执行性能关键的操作。同时,警告指出在其他操作中,还有额外的CPU指令集(如SSE、SSE2、SSE3、SSE4.1、SSE4.2、AVX2、AVX512F、AVX512_VNNI、FMA)是可以启用的,但目前还没有在TensorFlow编译时启用。


处理方法:在开头加上下面两句即可

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
JavaScript