一、什么是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