RuBP17 / AlphaDou

A Doudizhu reinforcement learning AI
GNU General Public License v3.0
4 stars 4 forks source link

CUDNN_STATUS_NOT_SUPPORTED #6

Open rubbyzhang opened 4 hours ago

rubbyzhang commented 4 hours ago

/root/miniconda3/envs/alphadou/lib/python3.12/site-packages/torch/nn/modules/conv.py:306: UserWarning: Plan failed with a cudnnException: CUDNN_BACKEND_EXECUTION_PLAN_DESCRIPTOR: cudnnFinalize Descriptor Failed cudnn_status: CUDNN_STATUS_NOT_SUPPORTED (Triggered internally at /opt/conda/conda-bld/pytorch_1712608847532/work/aten/src/ATen/native/cudnn/Conv_v8.cpp:919.) return F.conv1d(input, weight, bias, self.stride,

ubantu22.04 系统,4090D, 驱动550.67 pytorch 版本 2.3 cuda版本12.1 cuDNN8.9 . 会出现这个问题,后来尝试过 2.1 + 11.8 /12.1 依然存在这个问题,请教下解决思路

RuBP17 commented 3 hours ago

cuDNN的版本和cuda或者pytorch不匹配,可以去pytorch官网上看看。

There may be a mismatch between the version of cuDNN and the version of cuda/pytorch.