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