I am trying the LeNet5 on theano using cuda 8.0 windows 10 GTX 770M, and it seems the params cannot be updated correctedly. When I import theano, I will get this error:
Can not use cuDNN on context None: cannot compile with cuDNN. We got this error:
c:\users\yanjun~1\appdata\local\temp\try_flags_8n7fhy.c:4:19: fatal error: cudnn.h: No such file or directory
compilation terminated.
Mapped name None to device cuda: GeForce GTX 770M (0000:01:00.0)
I have installed cudnn 5 to my cuda 8.0 by copying the files to "NVIDIA GPU Computing Toolkit\CUDA\v8.0", but still the same error shows.
Is it because theano cannot compile the cudnn, both conv2d
and pool_2d
cannot work properly? As result, my error rate keeps around 90% while training. Did I install cudnn correctly? I just followed the instructions on theano's documentation, but no .so* file in the zip.
I've tried the cpu mode, it can be trained properly but really really slow.
Hope you guys could help me out of this where I am stuck for servel days.
Many thanks!