Theano很少使用GPU

剛開始接觸Theano和我的GTX 960，但在整個幾分鐘訓練的網絡（特別是複發性迴歸）和GPU使用高峯時達不到一秒鐘，也許兩個到三次花。有沒有辦法迫使它在整個時間內實際使用GPU？Theano很少使用GPU

似乎需要一個骯髒的很長一段時間沒有太多的數據。

2016-03-08 MKennedy

什麼是真正發生在你的代碼是，大部分的時間都花在獲取數據到和關閉GPU，但任何GPU顯示器將顯示該GPU是「不使用」在此期間，即使什麼它正在做的是加載數據。您無法避免這樣做，除了明智地將數據加載到GPU上的方式和時間。你所描述的尖峯是完全正常的 - 也就是說，當GPU實際進行計算時。沒有什麼可以做的！

來源

2016-03-08 21:29:54 mprat

在如你所描述的這種情況，在數據傳輸占主導地位的計算時間，它可能會更快地擺在首位在CPU上進行訓練。 – cfh

不一定 - 您仍然需要將數據導入GPU。耗時最長的事情是從磁盤讀取文件，然後將它們放在需要在電路板上物理存放的位置。 – mprat

我來自MatLab NN背景，幾乎瞬間將數據加載到GPU中並沒有問題。我不懷疑分鐘的延遲是由具有從RAM複製數據到有1.5Mb我GPU造成... – MKennedy

Theano很少使用GPU

回答

相關問題