WSS_ 2023-02-16 11:53 采纳率: 37.5%
浏览 27
已结题

python深度学习代码报错

请问这个地方的代码为什么一直报错?
报错位置第三行
错误:RuntimeError: All input tensors must be on the same device. Received cuda:0 and cpu
已知:x_c,x_p和y存储位置均为cuda:0

from inspect import Parameter
n = x_c.shape[0]
A = torch.cat([x_c,torch.ones(n, 1)], dim=1)
B = torch.cat([x_p,torch.ones(n, 1)], dim=1)
C = torch.cat([y,torch.ones(n, 1)], dim=1)
  • 写回答

2条回答 默认 最新

  • 元气少女缘结神 2023-02-16 12:10
    关注

    因为cat的两个变量一个在gpu一个在cpu,所以无法cat,将两个都移到gpu就行了。加上:

    device = torch.device(‘cuda:0’)
    在cpu变量后加个to(device)
    如torch.ones(n, 1).to(device) 即A = torch.cat([x_c,torch.ones(n, 1).to(device)], dim=1)

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论 编辑记录
查看更多回答(1条)

报告相同问题?

问题事件

  • 系统已结题 2月24日
  • 已采纳回答 2月16日
  • 创建了问题 2月16日

悬赏问题

  • ¥15 如何用QDomDocument读取内容为空格的xml数据
  • ¥15 请阅读下面代码,帮我修改下代码
  • ¥15 关于#microsoft#的问题:电脑启动后不显示桌面图标和窗口,除任务栏外无法操作任何东西
  • ¥15 如何输入百度,显示本地下载的html文件页面,地址栏还显示百度的地址
  • ¥15 通过kinect制作换装程序但是服装不贴合(标签-ar)
  • ¥20 matlab如何绘制三维瀑布图
  • ¥15 关于用abap来解决动态规划的问题,但是要求输出索引值,这个是难点
  • ¥15 在ISIS中什么是IP从地址
  • ¥15 压测时,并发量过高时,响应时间出现尖刺
  • ¥15 关于vmprotect3.8.4虚拟文件一项