当前位置: 首页 > 工具软件 > Numpy.GPU > 使用案例 >

AttributeError: ‘numpy.ndarray‘ object has no attribute ‘cuda‘

诸葛令
2023-12-01

问题描述:利用pytorch进行模型训练时,在加载数据后出现如下报错


inputs, gts = inputs.cuda(), gts.cuda()


AttributeError: ‘numpy.ndarray‘ object has no attribute ‘cuda‘

问题解决
这种类型的错误都是由于没有转换为tensor张量导致的,只需要注意使用cuda()前通过torch.tensor()转为tensor就可以了,例如这样:

inputs, gts = torch.tensor(inputs).cuda(), torch.tensor(gts).cuda()

参考博客:https://blog.csdn.net/qq_34124009/article/details/107541222

 类似资料: