解决RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

报错：解决RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED原因：1.可能cuda与python版本不一致详见：https://blog.csdn.net/qq_39938666/article/details/866114742.可能是你的GPU显存被占用光了，再运行就出错了。我的用cpu，代码就能跑了device = torch.d

人类高质量算法工程师

1240人浏览 · 2021-04-22 19:40:53

人类高质量算法工程师 · 2021-04-22 19:40:53 发布

报错：

解决RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

原因：
1.可能cuda与python版本不一致
详见：https://blog.csdn.net/qq_39938666/article/details/86611474
2.可能是你的GPU显存被占用光了，再运行就出错了。
我的用cpu，代码就能跑了

device = torch.device("cuda")

改为

device = torch.device("cpu")

天启AI社区

GitCode 天启AI是一款由 GitCode 团队打造的智能助手，基于先进的LLM（大语言模型）与多智能体 Agent 技术构建，致力于为用户提供高效、智能、多模态的创作与开发支持。它不仅支持自然语言对话，还具备处理文件、生成 PPT、撰写分析报告、开发 Web 应用等多项能力，真正做到“一句话，让 Al帮你完成复杂任务”。

更多推荐