问题描述
/home/luoxu/anaconda3/envs/vader_videocrafter_new/lib/python3.10/site-packages/torch/nn/modules/conv.py:605: 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_1712608935911/work/aten/src/ATen/native/cudnn/Conv_v8.cpp:919.) return F.conv3d(
解决办法
torch.backends.cudnn.enabled = False
在代码最前面加上这一行代码,也就是禁用cudnn,这样可能会损失性能,但大概率可以解决此问题。
具体的分析可见:cuDNN error: CUDNN_STATUS_NOT_SUPPORTED.This error may appear if you passed in a non-contiguous input - #5 by ptrblck - vision - PyTorch Forums