CUDA和cuDNN安装配置

您所在的位置:网站首页 论坛要下载吗 CUDA和cuDNN安装配置

CUDA和cuDNN安装配置

2024-07-15 13:14| 来源: 网络整理| 查看: 265

一、什么是CUDA和CUDNN

1 CUDA

CUDA(Compute Unified Device Architecture),是显卡厂商NVIDIA推出的运算平台。CUDA™是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。

CUDA 的架构下,一个程序分为两个部份:host 端和 device 端。Host 端是指在 CPU 上执行的部分,而 device 端则是在GPU芯片上执行的部分。 

2 CUDNN

CUDNN(CUDA Deep Neural Network library):是NVIDIA打造的针对深度神经网络的加速库,是一个用于深层神经网络的GPU加速库。

如果你要用GPU训练模型,cuDNN不是必须的,但是一般会采用这个加速库。

二、CUDA的下载及安装

1查询驱动版本

在开始菜单打开nv控制面板,然后根据右图查看自己的驱动版本

 2下载cuda

网址为CUDA Toolkit Archive | NVIDIA Developer

选择适合自己的版本进行下载

推荐本地下载安装

 3cuda安装

双击exe文件

点ok

选择自定义 

 

 全选

记住自己的安装位置

点击下一步就能进行安装

安装完成后检查是否有环境变量

一般都会有

没有的话手动配置一下

 

 三、cuDNN的下载及安装

 1cuDNN下载

网址为cuDNN Archive | NVIDIA Developer

选择与cuda版本对应的进行下载

注:下载需要注册一个账号

 下载好的cuDNN是压缩包,解压后直接复制到cuda目录下,有相同的文件直接替换

这是cudnn文件夹

 这是cuda文件夹

四、安装验证

运行cmd,输入nvcc --version 即可查看版本号

cd到安装目录下的 …\extras\demo_suite,然后分别执行bandwidthTest.exe和deviceQuery.exe

得到以下两张图

参考博客:

【CUDA】cuda安装 (windows版)_何为xl的博客-CSDN博客_cuda windows

 CUDA安装教程(超详细)_Billie使劲学的博客-CSDN博客_cuda安装

 深度学习之CUDA+CUDNN详细安装教程 - 知乎 (zhihu.com)



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3