成功解决RuntimeError: Expected all tensors to be on the same device, but found at least two devices。。。

最新推荐文章于 2024-08-15 09:32:42 发布

李卓璐

最新推荐文章于 2024-08-15 09:32:42 发布

阅读量395

点赞数 1

分类专栏：学习笔记文章标签： python

本文链接：https://blog.csdn.net/lzl2020lzl/article/details/139619340

版权

42 篇文章 13 订阅

订阅专栏

在yolov5中添加自己的模块时出现错误

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

PyTorch 中的张量（Tensors）是数据的主要表现形式，可以存在于计算设备 CPU或GPU上。当进行涉及多个张量的操作时，如加法、乘法或更复杂的神经网络层计算，所有这些张量都必须在同一设备上。

tensor1 = tensor1.cuda()  # 将张量转移到GPU上
tensor2 = tensor2.cpu()   # 将张量转移到CPU上

但是这种方法没有解决我的代码问题

#1.
if x.device != y.device:#可以省略
    y = y.to(x.device)  # 确保y在与x相同的设备上
#2.
y = self.xxx(y).to(x.device)

检查输入是否使用cuda

a = x.is_cuda
print(a)

关注

专栏目录