Unable to find valid cuDNN algorithm to run convolution
aalfi opened this issue · 2 comments
aalfi commented
Hello!
I am trying to train your example model.
I am getting this error. Please let me know if you can help
Thank you
From small-dataset.ipynb
trainer.fit_loader(train_loader,
val_loader=val_loader,
num_epoch=20,
verbose=1,
cuda_device=cuda_device)
derEitel commented
Do you have a matching version of CUDA and cuDNN installed for your pytorch version?
Please see here:
https://pytorch.org/get-started/previous-versions/
aalfi commented
Thank you for your clarification!