【E-14-Pytorch】RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 'mat1' in call to _th_addmm

一、出错误背景：

Pytorch 中想使用 CUDA 对程序计算进行加速

二、问题分析

错误的意思：object 的 device 类型期望得到的是 cuda 类型，但是实际上的类型确实 cpu 类型，在调用二分类交叉熵损失进行前向计算的时候

三、检查下面几点：

1. 模型是否放到了CUDA上 model = model.to(device) 或 model = model.cuda(device)
2. 输入数据是否放到了CUDA上 data = data.to(device) 或 data = data.cuda(device)
3. 模型内部新建的张量是否放到了CUDA上 p = torch.tensor([1]).to(device) 或 p = torch.tensor([1]).cuda(device)
   一般情况下应该是忘记了第三点，而根据提示也可以知道，在进行二分类交叉熵损失进行前向计算的过程中，存在没有放到cuda上的张量，找到他，fix it ！！！

其中：device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

四、解决方式