【深度学习】CUDA、cuDNN、pytorch安装

系统已有Anaconda环境，所以跳过这一步，按顺序安装CUDA、cuDNN、pytorch。

参考博客：2024最新超详细无痛安装PyTorch，包括下载地址、环境配置与各种避坑指南

一、安装CUDA

查询电脑显卡是否支持CUDA以及支持安装哪个版本

控制台输入nvidia-smi ，查看支持的CUDA版本

在官网下载可支持的版本，官网：https://developer.nvidia.com/cuda-toolkit-archive，我下载的12.0

选择精简版，一直下一步，直到安装完成。

控制台验证一下，输入nvcc -V，显示版本号即为成功

二、安装cuDNN

cuDNN要适配CUDA版本，下载地址：https://developer.nvidia.com/rdp/cudnn-archive，我下载的9.4

下载步骤和CUDA一样，选择精简版，一直点下一步，直到安装完成。

下载完成后将cuDNN的这三个文件夹复制到CUDA的目录下

文件夹名字相同不用管，会自动合并成一个。

三、下载pytorch

打开anaconda，创建一个虚拟环境
输入命令：conda create -n torchenv python=3.10
torchenv是虚拟环境的名字
python后是版本
不知道python版本可以控制台输入python --version查看

创建完成后激活
输入命令：conda activate torchenv

激活后直接输入命令，下载pytorch
输入命令：pip3 install torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

下载完成后如图，这一步耗时会比较长。

验证是否下载成功
1.输入python
2.输入import torch
3.输入torch.cuda.is_available()，结果返回True即为成功。