问小白 wenxiaobai
资讯
历史
科技
环境与自然
成长
游戏
财经
文学与艺术
美食
健康
家居
文化
情感
汽车
三农
军事
旅行
运动
教育
生活
星座命理

Windows下安装CUDA Toolkit和cuDNN的完整指南

创作时间:
作者:
@小白创作中心

Windows下安装CUDA Toolkit和cuDNN的完整指南

引用
CSDN
1.
https://blog.csdn.net/weixin_45558466/article/details/140059924

本文将详细介绍在Windows系统下安装CUDA Toolkit和cuDNN的完整步骤,包括版本判断、安装包下载、环境变量配置以及安装成功的检查方法。

1. 判断自己下载的CUDA版本

打开NVIDIA控制面板,查看当前系统支持的CUDA版本。例如,本文使用的是CUDA 12.5版本。

2. 下载CUDA Toolkit安装包

访问CUDA Toolkit官方下载页面:https://developer.nvidia.com/cuda-toolkit-archive,选择适合您系统的CUDA版本和配置进行下载。

3. 安装CUDA Toolkit

双击下载好的安装包开始安装:

  1. 选择解压路径,这个文件夹在安装完成后会被自动删除。

  2. 如果是第一次安装,建议全选所有选项;如果是多次安装,则只需选择第一个选项。

  3. 在CUDA选项卡中,取消勾选Visual Studio Integration。

  4. 设置安装路径时,确保该路径与解压路径不同,且需要记住这个路径,因为后续配置环境变量时会用到。

4. 配置环境变量

安装过程中会自动生成一些环境变量,如果没有自动生成,需要手动添加。

5. 检查CUDA Toolkit是否安装成功

打开命令提示符(Win键+R键,输入cmd回车),输入以下命令:

nvcc --version

如果能显示版本信息,说明CUDA Toolkit安装成功。接着输入:

set cuda

可以查看CUDA相关的环境变量设置。

6. 配置cuDNN

访问cuDNN官方下载页面:https://developer.nvidia.com/rdp/cudnn-archive,注册并下载与CUDA Toolkit相同版本的cuDNN。

下载后解压,将cuDNN文件夹下的includelibbin三个文件夹中的文件复制到CUDA Toolkit的安装目录下的对应同名文件夹中。

7. 配置环境变量

在系统环境变量的Path中添加以下四个路径:

  • CUDA Toolkit安装目录下的lib
  • CUDA Toolkit安装目录下的include
  • CUDA Toolkit安装目录下的bin
  • CUDA Toolkit安装目录下的libnvvp

8. 检查cuDNN是否安装成功

进入CUDA Toolkit安装目录下的extras\demo_suite文件夹,运行以下命令:

.\bandwidthTest.exe
.\deviceQuery.exe

如果显示安装成功,说明cuDNN已经正确安装。

参考资料

  • Windows CUDA安装教程(超详细)-CSDN博客
© 2023 北京元石科技有限公司 ◎ 京公网安备 11010802042949号