RuntimeError: "addmm_cuda" not implemented for 'Short'
时间: 2023-10-12 08:03:21 浏览: 339
onnxruntime windows下cuda驱动文件
这个错误通常是由于在使用 PyTorch 的 CUDA 版本时,数据类型不匹配导致的。在 PyTorch 中,有些操作是只支持特定的数据类型的,如果数据类型不匹配,就会出现这个错误。
根据错误提示信息,你使用了一个数据类型为 Short 的张量,而某个操作(addmm_cuda)没有实现对于这种数据类型的支持。你可以尝试将数据类型转换为其他支持的类型,比如 Float 或 Double,或者检查你的代码是否有其他错误。如果需要使用 Short 类型的张量,你可以尝试使用 CPU 版本的 PyTorch 运行代码。
阅读全文