引言:罗里吧嗦,多说无益!!!
1.错误:
error:RuntimeError:Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
reason:(小琼说一下自己的错误)出现在此错误的原因在于将网络使用GPU训练,但是只是部分训练用GPU,一部分使用CPU,所以会出现这种情况。
solve:将网络修改为使用GPU训练,同时所有相关的都修为使用GPG
训练。代码:
model=model.to(device)
说完了。
2.结束
GAME OVER
如果理解错误,欢迎大家批评,及时纠正小琼的错误哦,真心感谢你的纠正!!!
如果小伙伴梦有疑问欢迎在评论区留言哦!!!
如果感觉不错的话!点个赞呗,不用客气!(O(∩_∩)O哈哈~)
欢迎和小伙伴梦一起学习,共同努力,加油!!!
图片来自网络(如有侵权,请与唔联系)
#see you next time!!!
Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
最新推荐文章于 2024-06-12 10:28:02 发布