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