问题描述
/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