首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >专栏 >pytorch2.3.0有警告UserWarning: Plan failed with a cudnnException: CUDNN_BACKEND_EXECUTION_PLAN_

pytorch2.3.0有警告UserWarning: Plan failed with a cudnnException: CUDNN_BACKEND_EXECUTION_PLAN_

作者头像
云未归来
发布2025-07-17 20:07:44
发布2025-07-17 20:07:44
1100
举报

警告如下:

首先python里面警告一般不用管这个也一样,但是一定要消除这个警告需要如下解决方法:

方法一: 在代码中加入以下代码:

  torch.backends.cudnn.enabled=False 运行 禁用cudnn计算,报错消失,但是运行速度会变得很慢。

方法二:

减小batchsize大小。

方法三(推荐):

查看自己的pytorch 版本是不是2.3.0, 在 pytorch 2.3.0 版本中,即使没有抛出异常,它也会打印这个不需要的警告:可以参阅https://github.com/pytorch/pytorch/pull/125790

如果你想摆脱这个警告,可以恢复到 torch 2.2.2(然后你还必须将 torchvision 恢复到 0.17.2):

pip3 install torchvision==0.17.2 pip3 install torch==2.2.2 如果嫌弃版本太低将torch版本升级到>=2.3.1即可消除警告

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2025-05-11,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档