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安裝的更多詳細信息。