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

您可能还喜欢...