1. 程式人生 > >GPU 版tensorflow在訓練CIFAR資料集時GPU 沒有參與計算/出現假宕機狀態/CPU滿負荷運轉

GPU 版tensorflow在訓練CIFAR資料集時GPU 沒有參與計算/出現假宕機狀態/CPU滿負荷運轉

程式碼:《TensorFlow實戰》黃文堅著,第5.3節 TensorFlow 實現進階的卷積網路,所用資料集為CIFAR-10

出現問題的電腦型號:thinkpad T470; 顯示卡: GeForce 940mx

問題描述:書中提供的程式碼之前在不同的桌上型電腦上執行都沒有問題,後來在筆記本執行的時候發現CPU滿負荷運轉,而GPU沒有參與計算。開始以為是顯示卡相關驅動沒有安裝好,在對顯示卡驅動進行了幾番解除安裝和安裝後,TensorFlow終於從CPU負載下的緩慢計算變為了徹底不計算。。。執行程式後,一直停留列印顯示卡資訊那一步,後來才知道在呼叫CIFAR資料時預設是用的GPU,這回大大降低呼叫的速度,在呼叫CIFAR讀取程式的時候增加with tf.device('/cpu:0'),即指定用cpu即可解決: