WSS_ 2023-02-16 03:53 采纳率: 33.3%
浏览 29
已结题

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 04: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月23日
  • 已采纳回答 2月16日
  • 创建了问题 2月16日
手机看
程序员都在用的中文IT技术交流社区

程序员都在用的中文IT技术交流社区

专业的中文 IT 技术社区,与千万技术人共成长

专业的中文 IT 技术社区,与千万技术人共成长

关注【CSDN】视频号,行业资讯、技术分享精彩不断,直播好礼送不停!

关注【CSDN】视频号,行业资讯、技术分享精彩不断,直播好礼送不停!

客服 返回
顶部