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利用率降低。


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM