cuda-cudnn-pytorch 安装教程

cuda-cudnn-pytorch 安装教程

CUDA 安装

1. 使用 nvidia-smi 查看自己的 nvidia GPU CUDA 驱动版本

2. 去官网下载 CUDA 包。

https://developer.nvidia.com/cuda-toolkit-archive 选择本地安装 注意，下载的CUDA包的版本号不能大于自己GPU CUDA的版本。这是官方给的建议。 https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

3. 安装 CUDA 我们进行自定义安装。自定义安装这里可以进行一些设置，否则有可能会导致安装失败，首先组件这里取勾GeForce Experience。 然后， 我这里安装的时候，必须Visual Studio Integration也需要取消勾选。 点开Driver comonents，Display Driver这一行，前面显示的是Cuda本身包含的驱动版本是411.31 如果你电脑目前安装的驱动版本号新于Cuda本身自带的驱动版本号，那一定要把这个勾去掉。否则会安装失败(相同或小于的话，就不用去取勾了)

安装完成后，使用 nvcc -V 命令测试是否安装成功

cudnn 安装

cudnn 需要登录下载(科学上网)。 1. 官网下载：https://developer.nvidia.com/rdp/cudnn-archive。 下载的时候尽量与 CUDA 版本对应。

1. 解压文件夹，将解压后的文件夹下的文件拷贝到cuda安装目录下，与之相对应的文件夹下。
2. 添加系统环境变量

PyTorch 安装

1. 官网下载，找到自己对应的 cuda-pytorch

   https://pytorch.org/get-started/previous-versions/

2. 使用 pip 或 conda 安装 pytorch。安装的时候推荐使用虚拟环境，以避免依赖包冲突。