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

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

你可能感兴趣的:(代码,cuda,深度学习)