安装了最新版的cuda11.5+cudnn8.3.0后print(torch.cuda.is_available())结果却是False,查了一下原来安装的是CPU版pytorch,而不是GPU版。
费了一大番周折后发现原来我的源里都找不到cuda11.5+cudnn8.3.0版的pytorch,所以自动给我装了CPU版。从清华源里找到了pytorch-1.10.1-py3.8_cuda11.3_cudnn8_0.tar.bz2,所以就照这个下载吧,安装了cuda11.3后又发现找不到cuda11.3对应的cudnn8_0,只找到了cudnn-11.3-windows-x64-v8.2.0.53,再重新卸载安装cuda真是太麻烦了,先这么着试试吧,没想到竟然可以用!
总结一下,本次安装的是:
cuda_11.3.0_win10_network.exe
cudnn-11.3-windows-x64-v8.2.0.53
pytorch-1.10.1-py3.8_cuda11.3_cudnn8_0.tar.bz2
print(torch.cuda.is_available()),终于出现True了!!!