安装cuda、cudnn、Pytorch（用cuda和cudnn加速计算）

目录

写在前面

安装cuda

1、了解cuda版本

2、安装cuda

3、测试是否安装成功

安装cudnn

1、安装cuda后还要安装cudnn？

2、cudnn下载

安装pytorch-gpu

1、配置新的虚拟环境 

2、安装Pytorch

3、小插曲&安装方式二

4、检测是否安装成功

5、增加到Pycharm的编译器里

参考 

---

写在前面

最近几个月都在忙着毕业的事，好一阵子没写代码了。今天准备跑个demo，发现报错 AssertionError: Torch not compiled with CUDA enabled

不知道啥情况，因为之前有cuda环境，能用gpu加速，看这个报错信息应该是Pytorch下没有可用的cuda，不知道咋没了。因而决定从头开始配置一下环境，以此博文记录一下~

---

安装cuda

1、了解cuda版本

打开英伟达控制面板 -> 左下角“系统信息” -> 组件

可以看到：我的电脑所能兼容的最高cuda版本为12.5

2、安装cuda

进入官网：CUDA Toolkit Archive | NVIDIA Developer，选择合适的版本进行安装

因为我的电脑所能兼容的最高cuda版本为12.5，所以我选择了12.4进行下载安装，同时我的电脑系统为win11，如下：

安装时，选择“自定义（高级）” -> 取消勾选CUDA下的Visual Studio Integration

我选择的安装位置为F:\NVIDIA GPU Computing Toolkit\CUDA\v12.4。安装完成后检查一下是否有环境变量，没有的话手动添加一下

3、测试是否安装成功

在cmd里输入nvcc -V，出现类似下图的界面就ok了~

---

安装cudnn

1、安装cuda后还要安装cudnn？

• CUDA：是一个通用的并行计算平台和编程模型，