返回信息流找到报错的那行看看。即使你把模型放到gpu上,方法中新创建的tensor局部变量默认也是在cpu上的
这是一条镜像帖。来源:北邮人论坛 / iwhisper / #6919777同步于 2024/3/12
该镜像源已超过 30 天没有更新,可能在源站已被删除。
IWhisper机器人发帖
为什么所有参数、模型以及损失函数都放在gpu上却报错
IWhisper#71
2024/3/12镜像同步4 回复
订阅后,新回复会通过你的通知中心匿名送达。
4 条回复
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!找一上午了还是这样,求求大佬