CUDA详细安装及环境配置——环境配置指南 – CUDA+cuDNN+PyTorch 安装

发布于:2025-04-02 ⋅ 阅读:(25) ⋅ 点赞:(0)

确定显卡支持的CUDA版本

在显卡驱动被正确安装的前提下,在命令行里输入

nvidia-smi.exe

效果如图所示
在这里插入图片描述

可以看到显示CUDA Version为12.7,说明该显卡最高支持到12.7,我这里就选择12.6的版本,你也可以选择更低的版本比如 12.5,12.4更低的版本

确定CUDA版本对应的cuDNN版本

在cudnn下载页面,我们cuda是12.6,这里就选择cuDNNV8.9.0版本的for CUDA11.x版本即可
在这里插入图片描述

好了三个安装版本都确定好了,现在开始一个个安装就行

安装CUDA

下载安装包,在NVIDIA官方网站即可下载,地址为:https://developer.nvidia.com/cuda-toolkit-archive
在这里插入图片描述

注意选择你的操作系统,什么版本的 我是Windows11 需要选择11,离线安装包还是在线安装包,省得麻烦就选择离线安装包了,选择好后,点击下面的Download按钮。
安装包下载好后,双击安装包进行安装
pic_ba57e34d.png
弹出这个,临时抽取文件放置位置的,直接点击ok、随后进入系统兼容性的检查,就是看你是否下错安装包了
在这里插入图片描述

没问题后会弹出许可协议
pic_09db9d37.png
直接点击同意并继续,弹出的安装选项,选择自定义
在这里插入图片描述

点击下一步
在这里插入图片描述

把CUDA选一下,还需要注意一点,如果 你前面没有安装vs,直接安装的这个,需要把CUDA里面的
Visual Studio Integration取消勾选,否则会安装不成功
在这里插入图片描述

需要记住安装的位置,后面需要用到
pic_26d39cf6.png
下一步,开始安装
在这里插入图片描述

点击下一步
在这里插入图片描述

这里提示了你选择的组件的安装情况,到这就安装好了,点击关闭。

检查是否安装成功

打开cmd,输入

nvcc -V

在这里插入图片描述

出现了你安装的CUDA的版本信息,说明安装成功了

安装cuDNN

下载安装包,在NVIDIA官方网站即可下载,地址为:
https://developer.nvidia.com/rdp/cudnn-archive
如没有NVIDIA开发者账号的话,就按照提示注册一个就好,再登录即可下载了。
下载下来是个压缩包,我的是cudnn-windows-x86_64-8.9.7.29_cuda12-archive.zip 直接解压缩,完成后点击去你能看到如下三个文件夹(bin、include、lib)
pic_34b9a072.png
把这三个文件夹的文件分别拷贝到CUDA安装目录对应的(bin、include、lib)文件夹中即可。CUDA的lib目录有x64 、Win32、cmake三个文件夹,拷到其中的x64这个文件夹中
pic_2fa9689e.png
然后进入该目录下的 extras/demo_suite/ 目录,在终端中依次输入以下命令:

./bandwidthTest
./deviceQuery

在这里插入图片描述
在这里插入图片描述

若均输出 Result = PASS,说明安装成功:

安装 PyTorch

然后,进入 PyTorch 官网,通过“Get Started -> Start Locally”定位到下载页:

https://pytorch.org/

在这里插入图片描述
在这里根据你使用的 Python 环境选择合适的安装方式,网页上会自动生成合适的安装命令。例如我使用 pip 管理我的 Python 环境,安装了 CUDA 12.x,就选择相应的选项。如果你使用 Anaconda 管理你的 Python 环境,你需要选择“Conda”选项。此处我使用以下代码安装:

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126