①安装cuda
②安装cudnn
③安装pytorch
安装cuda:查看gpu支持的cuda版本,右键进入NVIDIA控制面板->帮助->系统信息->组件,这里支持CUDA10.1
下载10.1版cuda
CUDA Toolkit Archivedeveloper.nvidia.com安装cuda:先安装Visual Studio,我这里装的是VS2019,否则会出现安装Visual Studio Integration报错,然后安装cuda,选择精简安装即可,通过在cmd中键入命令 nvcc -V测试是否安装成功。
安装cudnn:下载10.1版cudnn,解压复制到cudn安装目录:C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.1下,可以看到include文件夹下包含cudnn.h文件
NVIDIA cuDNNdeveloper.nvidia.com安装Pytorch
PyTorchpytorch.org安装成功