【踩坑日记23】UserWarning: Plan failed with a cudnnException: CUDNN_BACKEND_EXECUTION_PLAN_DESCRIPTOR

问题描述

/mnt/lab/XXX/anaconda3/envs/diffusion_ddpo/lib/python3.10/site-packages/torch/nn/modules/conv.py:456: 
UserWarning: Plan failed with a cudnnException: CUDNN_BACKEND_EXECUTION_PLAN_DESCRIPTOR: cudnnFinalize Descriptor Failed cudnn_status: CUDNN_STATUS_NOT_SUPPORTED (Triggered internally at ../aten/src/ATen/native/cudnn/Conv_v8.cpp:919.)
  return F.conv2d(input, weight, bias, self.stride,

问题分析

问题出在文件夹“anaconda3/envs/diffusion_ddpo/”中，是关于环境中的包的问题，通过查阅pytorch的github官网，发现是Pytorch 2.3.0的问题。

问题解决

• Pytorch 2.3.1已经解决该问题了，因此可以将Pytorch进行升级
• 将Pytorch降级至2.2.0

参考

UserWarning: Plan failed with a cudnnException