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利用率降低。
唯有热爱方能抵御岁月漫长。