问题在于下载的torch没有cuda,即你的电脑没有GPU,无法运行
解决办法:在程序开头添加下列代码,然后在出现.cuda()的地方换成.to(device)
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
问题在于下载的torch没有cuda,即你的电脑没有GPU,无法运行
解决办法:在程序开头添加下列代码,然后在出现.cuda()的地方换成.to(device)
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")