Tensorflow 导入非常慢，因为它正在寻找“cudart64_110.dll”。 我没有 gpu。有 cpu-only 版本吗？

Question

当我运行导入 tensorflow (2.8) 的脚本时，它会在运行我的脚本之前花费 10 多秒打印下面的消息。 我没有 gpu。

从我在这里可以找到的内容来看，v2 向上结合了 cpu/gpu 版本。 是否可以停止 gpu 支票？

2022-02-04 10:15:46.628917: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'cudart64_110.dll'; dlerror: cudart64_110.dll not found
2022-02-04 10:15:47.068570: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.

Answer 1

是的，您可以使用以下代码超越这些warnings (W) 或informational (I) 消息：

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

您还可以查看此链接以获取有关TensorFlow安装的更多详细信息。