search

uci-cbcl / UFold

MIT License
58 stars 26 forks source link

RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED #20

Open sfsdfd62 opened 1 year ago

sfsdfd62 commented 1 year ago

My version information is as below: cuDNN:8.5.0 cuda：11.7 pytorch=1.4.0 python=3.6.6 Ubuntu 20.04.4

yikang7720 commented 1 year ago

Try torch.backends.cudnn.enabled = False, it resolves my problem, but the execution becomes very slow.

sfsdfd62 commented 1 year ago

Try torch.backends.cudnn.enabled = False, it resolves my problem, but the execution becomes very slow.

I understood this situation. This is because I'm using GTX 3090, which only supports CUDA version>11.0 and pytorch >some certain release. But it seems that Pytorch used in this work is way too out-dated, so we may need to upgrade it.