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) 编辑