问题:AssertionError: Torch not compiled with CUDA enabled

问题在于下载的torch没有cuda,即你的电脑没有GPU,无法运行

解决办法:在程序开头添加下列代码,然后在出现.cuda()的地方换成.to(device)

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

问题:AssertionError: Torch not compiled with CUDA enabled_第1张图片

你可能感兴趣的:(python,深度学习,pytorch,人工智能)