2016-03-08 70 views
0

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

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

回答

1

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

+0

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

+0

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

+0

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