How to configure GPU for theano on Jetson TK1

Hi guys,

I want to use theano as my deep learning library and everything goes until I start to configure GPU on TK1.
My .theanorc file is shown:


[global]
device =cuda
floatX=floatX

[cuda]
root = /usr/local/cuda-6.5

[lib]
cnmem=0.8

however, when I import theano, I got the errors below:

ERROR (theano.gpuarray): Could not initialize pygpu, support disabled

I guess the reason is the missing of GPU, but I have no idea how to figure it out.
Thanks

Hi,

Thanks for your question.

Since we are not familiar with theano, could you provide the steps of installation so we can check this issue directly?

Hi Ting-Kuei,

Could you provide the steps of installation so we can check this issue directly?

Thanks

Has someone successfully installed pygpu on the TK1 to get GPU support in Theano on TK1? I believe that is what is required to start testing this. To install by building from source cmake 3.0 is required and I do not believe that is available on TK1/Ubuntu.

Hi,

We want to check this issue but we are not familiar with theano.
Could you share the steps to build theano that we can check if pygpu work or not on tegra?

Thanks.