问题描述

<code>/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.)code>

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