print(torch.cuda.is_available())
时间: 2023-11-13 09:45:06 浏览: 98
torch_scatter-2.0.7-cp36-cp36m-linux_x86_64whl.zip
print(torch.cuda.is_available())用于检查当前环境下是否可用GPU。根据引用的描述,当输出为False时,意味着当前环境没有可用的GPU。这可能是因为你下载的是pytorch的CPU版本而非GPU版本所导致的。为了解决这个问题,你可以按照引用中的提示,确保你下载并安装了正确的GPU版本。另外,你还可以通过检查CUDA版本和对应的cudatoolkit来确保你下载的pytorch与你的GPU兼容。
阅读全文