Background:
Use a graphics card in the ubuntu18.04
system geforce RTX 3090
to reproduce r2c
problem
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
Cause analysis:
The graphics card geforce RTX 3090
only supports versions of cuda11 and above.
Solution:
Update pytorch and CUDA versions:
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch