会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
在青青草原上抓羊
博客园
首页
新随笔
联系
订阅
管理
2022年3月18日
【深度学习】RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
摘要: 报错代码: if __name__ == '__main__': model = Perception(2, 3, 2).cuda() input = torch.randn(4, 2).cuda() output = model(input) # output = output.cuda() la
阅读全文
posted @ 2022-03-18 10:49 SeanSiyang
阅读(7635)
评论(0)
推荐(0)
编辑
公告