RuntimeError: “addmm_cuda“ not implemented for ‘Int‘

最新推荐文章于 2024-08-23 09:44:21 发布

Surupor

最新推荐文章于 2024-08-23 09:44:21 发布

阅读量7.4k

点赞数

分类专栏： pytorch Python编程人工智能原理文章标签：深度学习机器学习 python

本文链接：https://blog.csdn.net/sinat_38432876/article/details/126692360

版权

Python编程同时被 3 个专栏收录

5 篇文章 0 订阅

订阅专栏

pytorch

2 篇文章 0 订阅

订阅专栏

人工智能原理

2 篇文章 0 订阅

订阅专栏

本文探讨了在使用PyTorch时遇到CUDA不支持整数类型矩阵乘法的问题，通过示例代码展示了在GPU和CPU环境下运行的结果差异。当尝试在CUDA设备上进行整数矩阵乘法时，会抛出'addmm_cuda'未实现的错误。而当切换到CPU时，整数矩阵乘法可以正常执行。此问题提示我们，对于特定的运算，需要使用CUDA支持的数据类型来确保计算的顺利进行。

摘要由CSDN通过智能技术生成

cuda不支持整数的矩阵乘法

如下所示：

import torch
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(device)
a=torch.tensor([1,2,3],dtype=torch.int).to(device)
b=a.unsqueeze(dim=-1)@ a.unsqueeze(dim=0)
print(b)

结果

Traceback (most recent call last):
File "____mytest.py", line 5, in <module>
b=a.unsqueeze(dim=-1)@ a.unsqueeze(dim=0)
RuntimeError: "addmm_cuda" not implemented for 'Int'

但是当改成CPU时

import torch
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(device)
a=torch.tensor([1,2,3],dtype=torch.int)
b=a.unsqueeze(dim=-1)@ a.unsqueeze(dim=0)
print(b)

cuda
tensor([[1, 2, 3],
[2, 4, 6],
[3, 6, 9]], dtype=torch.int32)