背景:
在ubuntu18.04
系统使用显卡GeForce RTX 3090
复现R2C
问题
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
原因分析:
显卡GeForce RTX 3090
仅仅支持使用cuda11及以上
的版本。
解决方案:
更新pytorch、cuda版本:
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch
参考(解决方案汇总):RuntimeError:CuDNN error:CUDNN_STATUS_EXECUTION_FAILED