记录一下自己安装pytorch后显示torch.cuda.is_available()的解决过程。
1. 查看gpu的型号:
打开任务管理器,选择性能,选择GPU0(或GPU),会在界面显示gpu的型号
2. 查看cuda和cudnn的安装情况
在文件夹C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA下查看安装的cuda的版本,例如该文件夹下有v10.0,则代表安装的是v10.0版本的cuda.
在 cmd 中输入 nvidia-smi
,同时这个指令也可以查看 GPU 的一些信息,如果出现如下界面就说明 CUDA 安装成功了
3. 找到和cuda版本以及python版本匹配的离线安装包进行安装。在线安装容易安装成cpu版本的,或者连不上包管理服务器。
离线包下载地址
https://download.pytorch.org/whl/torch_stable.html
例如本人电脑上安装的是python3.7, cuda版本是10.0,则下载:cu100/torch-1.2.0-cp37-cp37m-win_amd64.whl cu100/torchvision-0.4.0-cp37-cp37m-win_amd64.whl 这两个文件