CUDA环境配置（自用）

我自己电脑的配置环境是Python 3.7+CUDA 12.1+Pytorch 2.51

一、查看自己电脑支持的最高的nvcc版本（win+R   cmd   nvidia-smi）,我这里最高支持12.3版本

二、下载合适的CUDA版本

在Pytorch官网查看对应的下载终端指令，在pycharm终端输入该指令运行即可

下载链接：https://developer.nvidia.com/cuda-toolkit-archive

下载中记得配置好cuda的环境变量

• 下载CUDA对应的版本号cudnn

下载链接：https://developer.nvidia.com/rdp/cudnn-archive#a-collapse896-120

看着选对应的版本就行

下载后将cudnn的三个文件夹bin、include、lib放到CUDA的安装路径下

以下是我的CUDA安装路径（可在编辑系统环境变量中查看）

• 验证是否配置成功CUDA

打开命令行 win+R   输入nvcc-V

• 在pychram进行安装

Pytorch官网：PyTorch

将下面的指令在pycharm的终端进行输入

如果是通过anacoda进行安装的，输入conda list 进行查看，是否安装成功。

如果是通过pip进行安装的，输入pip list查看，

如果以上查看有torch、torchaudio、torchvision三个list文件，则说明其安装成功，否则重新安装