Tensorflow不使用nvidia gpu，而是CPU利用率为100％

Posted 2023-01-12

我正在使用anaconda发行版来安装tensorflow gpu（conda create --name tf_gpu tensorflow-gpu）。我已经在anaconda导航器中创建了虚拟环境。在我训练模型时，cpu利用率为100，gpu利用率约为3-5％。

其他一切正常。

Tensorflow v 2.1.0tensorflow-gpu v 2.1.0

答案

您正在训练哪种模型，以及使用哪种脚本/ SDK？例如，deepspeech既具有在CPU上运行的'deepspeech'软件包，又具有利用GPU的'deepspeech-gpu'软件包。