PyTorch+CUDA+cudnn安装

1、安装CUDA

先查看一下电脑显卡对应的CUDA驱动版本

 

上图可以看到CUDA驱动版本为11.1.114，那就下载11.1版本的CUDA好了。（这里应该也不是必须对应，只是说有可能下载低版本的会有问题，没试过，下载对应版本的先跑起来再说）

下载地址：

https://developer.nvidia.com/cuda-toolkit-archive 

 下载好之后点击文件安装，会先解压到一个临时文件夹内，默认就行

 然后出现安装步骤

我这里就选这四个选项进行安装（没有问题），下面是安装路径

 到这里就安装好了，验证一下

上面就说明CUDA安装好了，下面就继续安装cudnn

2、cudnn安装

Cudnn的版本可以根据CUDA的版本确定，不用担心

Cudnn下载链接:

https://developer.nvidia.com/rdp/cudnn-archive

Cudnn下载好之后是个压缩包，将其解压后包含三个文件夹

Cudnn的安装其实就是把这三个文件夹与CUDA安装目录下同样文件名的文件夹合并

 

然后验证一下

 到这里cudnn也配置好了

3、安装pytorch-gpu

这里安装就没什么说的了，去官网一键安装就行

https://pytorch.org/get-started/locally/

 最新的版本不是想要的，去历史版本找到想要的

 官网有conda和pip两种方式安装。

这样就全部OK了。

测试一下

end.