RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

背景：

在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

版权声明：本文为qq_43218033原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。