Closed leej35 closed 6 years ago
DId you compile DyNet with cuDNN?
And also: DyNet only uses cuDNN for a small subset of the functions, including Conv2D. You can run a regular LSTM without cuDNN, so it's likely that cuDNN simply wasn't installed w/ DyNet.
Thanks for advices. I reinstalled Dynet with cuDNN path explicitly fed in and it works well now.
Hi, I tried to run this CNN on text demo at: https://github.com/neubig/nn4nlp2017-code/blob/master/05-cnn/cnn-class.py
When I run the code without GPU flag, it works without any problem. But when I run it with GPU flag, it gives me this error:
To confirm that CUDNN installation is okay, I ran these codes on language model (https://github.com/neubig/nn4nlp2017-code/blob/master/02-lm/nn-lm-batch.py) and LSTM-sentiment modeling (https://github.com/neubig/nn4nlp2017-code/blob/master/06-rnn/sentiment-lstm.py) with GPU flag and these were executed without problem.
So I suspect probably there can be some issue on CUDNN backed convolution layer of Dynet.
FYI, I use Ubuntu 17.04 with CUDA 8.0 and CuDNN 5.1.
Thanks, Jeongmin