- 最气人的是torch.load默认是在GPU0上的, 需要map一下
- torch.load(‘trained.pth’, map_location={‘cuda:0’: ‘cuda:1’})
pytorch的那些坑
猜你喜欢
转载自blog.csdn.net/landing_guy_/article/details/121481564
今日推荐
周排行