2016-06-11 64 views
3

我是Theano的新手,嘗試一些例子。爲什麼theano跑得這麼慢?

import numpy 
import theano.tensor as T 
from theano import function 
import datetime 
print datetime.datetime.now() 
x = T.dscalar('x') 
y = T.dscalar('y') 
z = x + y 
f = function([x, y], z) 
print f(2, 3) 
print numpy.allclose(f(16.3, 12.1), 28.4) 
print datetime.datetime.now() 

然後花了15分鐘來運行它。我使用的是2GB內存,並且沒有多個進程同時運行。

+0

你可以分解它看看哪一步花了時間?可能是'f = function(...)',編譯發生的地方,但很好確保。 – user20160

+0

我運行了相同的一段代碼,它在一秒之內運行。你有沒有在你的機器上更新NumPy&Theano的版本? – KartikKannapur

+0

f =函數([x,y],z)佔用95%以上的時間,其他一切運行平穩 –

回答

3

先檢查Theano標誌。

如果您沒有運行:THEANO_FLAGS = mode = FAST_RUN或者沒有運行默認標誌或者如果您更改.theanorc,可能需要一些時間。

-

然而,讀到這裏:

http://deeplearning.net/software/theano/tutorial/using_gpu.html

你還可以看到更多關於Theano旗幟在這裏:

http://deeplearning.net/software/theano/library/config.html

因爲你必須從運行IDE,您將不得不編輯.theanorc

作爲對Theano鏈接如上所述:

「 ,缺省值爲$ HOME/.theanorc。在Windows上,它默認爲$ HOME/.theanorc:$ HOME/.theanorc.txt以使Windows用戶的生活更輕鬆。 「

確切標誌是這樣的:

config.mode

字符串值: '模式', 'ProfileMode'(廢棄), 'DebugMode', 'FAST_RUN', 'FAST_COMPILE'

如果這並不幫助,一定要更新Theano出血邊緣編輯您的問題與theanorc SE ttings

PIP安裝--upgrade --no-DEPS混帳混帳+://github.com/Theano/Theano.git


您還可以使用OpenMP的使用額外的線程/核:

THEANO_FLAGS =模式= FAST_RUN THEANO_FLAGS = '的openmp =真' OMP_NUM_THREADS = 4蟒x.py