解决yoloV7报错：indices should be either on cpu or on the same device as the indexed tensor (cpu)

已于 2023-11-17 15:53:59 修改
阅读量424 收藏 3
点赞数 2
文章标签： YOLO
于 2023-11-17 15:51:15 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/wl30130611/article/details/134464680
版权
文章指导如何在loss.py文件中针对CUDA和CPU设备进行适当调整，包括修改torch.device的使用、将ones转换为相应设备的tensor以及确保matching_matrix初始化使用设备。
摘要由CSDN通过智能技术生成

step1 在loss.py文件导包后添加

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

step2  添加to(device）

修改：from_which_layer.append(torch.ones(size=(len(b),)) * i)
修改为：from_which_layer.append((torch.ones(size=(len(b),)) * i).to(device))

备注：loss.py文件中有多处，每处都需要修改

step2  添加device=device

修改：matching_matrix = torch.zeros_like(cost)
修改为：matching_matrix = torch.zeros_like(cost, device=device)

备注：loss.py文件中有多处，每处都需要修改

就爱学点yolo
关注
All input tensors must be on the same device
jacke121的专栏
03-31 6877
All input tensors must be on the same device RuntimeError: All input tensors must be on the same device. Received cuda:0 and cpu#18 RuntimeError: All input tensors must be on the same device. Received cuda:2 and cuda:0 pytorch在进行计算时， 张量必须在同一设备，都在cp..
yolov7报错 indices should be either on cpu or on the same device as the indexed tensor (cpu)解决
quyingzhe0217的博客
09-28 1669
yolov7 报错RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu) 解决
RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)
最新发布
weixin_44012667的博客
07-28 370
RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)-CSDN博客。
yolov7 RuntimeError:indices should be either on cpu or on the same device as the indexed tensor(cpu)
m0_55217834的博客
12-28 4390
在loss.py的第742行，将原先的device='cpu'替换为device='cuda:0'即可解决问题
YOLO7报错：indices should be either on cpu or on the same device as the indexed tensor (cpu)
热门推荐
原来是肖哈哈的博客
12-07 1万+
当我们的数据有部分在GPU上运行，有部分在CPU上运行时会报这个错， 一般有GPU的话都会选择在GPU上面跑模型，但要注意将其他定义的对象也放在GPU上面，否则应该默认是在CPU上面。 如图所示， x是从GPU中传过来的，但idx不是，idx是我们自己生成的，它默认放在CPU中，所以我们需要也把它放到GPU中，解决方法：加 .to(DEVICE) 其中DEVICE已定义。具体解决办法： 在loss.py文件中增加下图中第一行，修改下面二三行 1.device = targets.device 2.
YoloV7 RuntimeError: indices should be either on cpu or on the same device...
qq_41747040的博客
12-02 1149
yolov7
在使用yolo训练的时候出现以下错误RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)
04-29
要解决这个问题，可以将索引张量（indices）移动到与被索引的张量（indexed tensor）相同的设备上，例如： ```python # 将索引张量（indices）移动到与被索引的张量（indexed tensor）相同的设备上 indices = ...
indices should be either on cpu or on the same device as the indexed tensor (cpu)
09-24
indices应该要么放在CPU上，要么放在与被索引的张量相同的设备上（CPU）。这是因为索引操作需要访问被索引的张量的数据，并且被索引的张量的数据存储在特定的设备上。如果索引和被索引的张量在不同的设备上，那么就...
YOLOV7: indices should be either on cpu or on the same device as the indexed tensor (cpu)
u012254599的博客
02-12 635
【代码】YOLOV7: indices should be either on cpu or on the same device as the indexed tensor (cpu)
YOLO7 indices should be either on cpu or on the same device as the indexed tensor (cpu)
qq_40940671的博客
11-01 4831
YOLOv7
解决报错RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)
weixin_48829789的博客
03-17 832
在复现论文DPGN: Distribution Propagation Graph Network for Few-shot Learning的过程中遇到的问题。解决方案：代码看起来希望索引在cpu上，那么就在定义它的地方修改device到cpu上就好了。
评论 4
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值