TensorFlow使用GPU训练时CPU占用率100%而GPU占用率很低

在训练keras时,发现不使用GPU进行计算,而是采用CPU进行计算,导致计算速度很慢。

用如下代码可检测tensorflow的能使用设备情况:

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices()) 

查看是否只有CPU可用,发现不是,有GPU可用,但是为什么GPU利用率极低并且只有一个GPU在使用,另一个GPU利用率为0,

发现在启动时有一行报错:

Could not load dynamic library 'cudart64_101.dll'; dlerror: cudart64_101.dll not found

但是并没有在意,发生这个错误的原因在于新建的虚拟环境中没有这个文件,将环境切回base后GPU利用率升高,CPU利用率降低。

唯有热爱方能抵御岁月漫长。
原文地址:https://www.cnblogs.com/syq816/p/14519691.html