当前位置:首页 » 《我的小黑屋》 » 正文

【快速解决】UserWarning: Plan failed with a cudnnException: CUDNN_BACKEND_EXECUTION_PLAN_DESCRIPTOR

16 人参与  2024年10月20日 11:20  分类 : 《我的小黑屋》  评论

点击全文阅读


问题描述

/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


点击全文阅读


本文链接:http://zhangshiyu.com/post/174484.html

<< 上一篇 下一篇 >>

  • 评论(0)
  • 赞助本站

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

关于我们 | 我要投稿 | 免责申明

Copyright © 2020-2022 ZhangShiYu.com Rights Reserved.豫ICP备2022013469号-1