用GPU跑pytorch搭建的LSTM的时候出现了奇怪的报错

Question

简单来说就是PyTorch没有为CUDA准备名为“aten:：mkldnn_rnn_layer”的函数，求解该怎么搞，现在只能暂时先用CPU跑了

Answer 1

引自强大而免费的微信小程序：皆我百晓生。如有帮助，还望采纳。

---

根据错误信息，该问题是因为 PyTorch 没有提供名为 "aten::mkldnn_rnn_layer" 的 CUDA 函数，导致无法在 GPU 上运行。

为了解决这个问题，你可以考虑以下几个步骤：

1. 确保你的 PyTorch 版本是最新的。如果不是最新版本，尝试升级 PyTorch 到最新版本，可能已经修复了这个问题。

2. 确保你的 CUDA 版本与 PyTorch 版本兼容。检查你的 CUDA 版本是否在 PyTorch 文档中所列兼容的范围内。

3. 检查你的代码中是否有定义自定义的 RNN 层，特别是 "aten::mkldnn_rnn_layer"。如果存在这样的层，请尝试使用 PyTorch 提供的标准 RNN 层，如 LSTMCell 或 LSTM。

4. 如果以上步骤仍然无法解决问题，可能是由于你的 PyTorch 安装没有包含所需的 CUDA 模块。你可以尝试重新安装 PyTorch，并确保在安装过程中选择了正确的 CUDA 支持选项。

如果以上步骤仍然无法解决问题，建议你编辑提问将报错信息以代码形式贴上来，以便我们能更精准地帮助排查问题