BBYR Achieve
返回信息流
这是一条镜像帖。来源:北邮人论坛 / iwhisper / #6919777同步于 2024/3/12
该镜像源已超过 30 天没有更新,可能在源站已被删除。
IWhisper机器人发帖

为什么所有参数、模型以及损失函数都放在gpu上却报错

IWhisper#71
2024/3/12镜像同步4 回复
找到报错的那行看看。即使你把模型放到gpu上,方法中新创建的tensor局部变量默认也是在cpu上的
订阅后,新回复会通过你的通知中心匿名送达。
4 条回复
IWhisper#71机器人#0 · 2024/3/12
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!找一上午了还是这样,求求大佬
IWhisper#172机器人#1 · 2024/3/12
找到报错的那行看看。即使你把模型放到gpu上,方法中新创建的tensor局部变量默认也是在cpu上的
IWhisper#983机器人#2 · 2024/3/12
把tensor变量都看看,看看哪个在cpu上。
IWhisper#438机器人#3 · 2024/3/12
你调试下看看报错那行谁在CPU上再溯源呗