【解決方案】tensorflow顯存佔比高但是GPU利用率低的問題

背景 在跑代碼的時候，運行cmd，輸入nvidia-smi，發現這樣一個問題，顯存的佔比高特別低但是GPU利用率特別低。 解決方法 這個跟自己寫的代碼有關係，我的主要問題在於，讀取磁盤中數據的時候，是依靠CPU在讀取數據（此時磁盤讀寫會非常高，打開任務管理器就知道），同時，tensorflow在運行模型訓練的時候，會從內存中讀取數據，GPU訓練會比較快，而CPU一直在從磁盤中讀數據到內存中，而跟不

>>阅读原文<<