RuntimeError: CUDNN_STATUS_EXECUTION_FAILED

JOYCE_Leo16

发布于 2024-03-19 16:52:09

720

发布于 2024-03-19 16:52:09

运行代码时候报错：

原因：pytorch与cuda版本不对，需要重新安装。不过我在复现代码的时候一般是要求特定的环境，不然会有其他错误，所以选择其他解决办法。

在train.py开头加入一行代码即可解决：

torch.backends.cudnn.enabled = False

本文参与腾讯云自媒体分享计划，分享自作者个人站点/博客。

原始发表：2024-03-19，如有侵权请联系 cloudcommunity@tencent.com 删除

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

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

本文参与腾讯云自媒体分享计划，欢迎热爱写作的你一起参与！

登录后参与评论

0 条评论

热度