我正在嘗試使用我的 GPU NVIDIA GEFORCE GTX 1050 和 tensorflow 來訓(xùn)練神經(jīng)網(wǎng)絡(luò)。我嘗試了不同神經(jīng)網(wǎng)絡(luò)的不同代碼示例，結(jié)果總是 GPU 利用率為 8%，計算速度比 CPU 慢得多。在此屏幕中，您可以查看列車期間的利用率。任何解決方案？
2 回答
臨摹微笑
TA貢獻(xiàn)1982條經(jīng)驗 獲得超2個贊
有時 GPU 利用率受 HDD 速度的限制。特別是當(dāng)訓(xùn)練數(shù)據(jù)不適合 RAM 并且包含數(shù)百萬張圖像時，我觀察到一次訓(xùn)練與從磁盤讀取所有數(shù)據(jù)所需的時間完全相同。在我的情況下，使用 ssd 甚至 nvm 驅(qū)動器更改 hdd 會按比例增加 tensorflow 訓(xùn)練速度 5 倍。
添加回答
舉報
0/150
提交
取消