RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking arugment for argument mat1 in method wrapper_addmm)
说至少有两个设备,cup 和 cuda 0
其实就加了一个MLP,然后想测一下,结果就报这个错误
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking arugment for argument mat1 in method wrapper_addmm)
方法是:
self.mlp1 = MLP(2048) #举例:加的这个代码
在这个代码后面加上下面这句:
self.mlp1.cuda()
问题解决!
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and
最新推荐文章于 2024-08-08 13:41:47 发布