这个屌丝很懒，什么也没留下！

Pytorch，GPU改成CPU_model(s.unsqueeze(0).cuda()).cpu()

作者：我家自动化 | 2024-07-19 12:30:09

model(s.unsqueeze(0).cuda()).cpu()

1、修改torch.load

torch.load(opt.model, map_location='cpu')
1

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
.cuda()  -->(修改为) .to(device)
1
2

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/我家自动化/article/detail/851505