cuDNN launch failure : input shape ([306,1,16,29]) about voca HOT 5 CLOSED

Hi,

can you please provide some information about the cuda and cudnn version that you are using?

from voca.

Hi,

sorry for getting back to you late. I am using cuda 9.0 and cudnn 7. ^_^

from voca.

The code was tested with cuda 9.0 and cudnn 7.1
Sorry but I don't know what could cause your error

from voca.

Hi,

Thank you so much for your reply ^_^
The problem is caused by gpu memory, I added the following code in inference.py, it works prefectly!
<
os.environ['CUDA_VISIBLE_DEVICES']='2'

config = tf.ConfigProto()
config.gpu_options.allow_growth=True
sess = tf.Session(config=config)

from voca.

Hi, great that it works now and thanks a lot for the feedback

from voca.