pytorch GPU安装

在安装GPU的时候，踩坑不少，现安装成功，特分享以记录

一、查看本机GPU版本

桌面–右键—打开NIADIA控制面板—帮助—系统信息—组件—NIVDIA.DLL 查看cuda最高可安装版本

二、官网下载与系统匹配的cuda

https://www.geforce.com/hardware/technology/cuda/supported-gpus
亦可以百度搜索对应cuda下载，然后安装
如：https://developer.nvidia.com/cuda-11.3.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local
该地址下载的是window 10下的cuda11.3版本

三、下载并安装与cuda匹配的cudnn

https://developer.nvidia.com/rdp/cudnn-download
需要注册，

四、下载与cuda匹配的pytorch

https://pytorch.org/
官网查看与已下载cuda匹配的pytorch,并选择对应的命令
如果是conda。。。命令，如：
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
以上命令安装的是与cuda11.3匹配的pytorch命令，需要在anaconda下安装（已经安装了anaconda）
对应版本的具体安装命令可以在https://pytorch.org/get-started/previous-versions/查询
最终测试cuda安装成功