(前提是要有支持CUDA的显卡,基本都是英伟达公司推出的NVIDIA显卡,具体型号需要到官网上查询)
概念:
CUDA(Compute Unified Device Architecture,统一计算架构)是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。
cuDNN 是基于CUDA的用于深度神经网络的GPU加速库,有了它才能在GPU上完成深度学习的计算。cuDNN是插入式设计,即所谓安装只需把cuDNN文件复制到CUDA对应文件夹就可以了。
安装步骤:
1.官网下载CUDA
查看本机支持的CUDA版本
英伟达CUDA Toolkit历史工具包的网址:
https://developer.nvidia.com/cuda-toolkit-archive
2.官网下载
cuDNN官网
https://developer.nvidia.com/cudnn
注意与CUDA版本相匹配!
(需要有nvidia账号)
3.安装程序
选择自定义
全选
安装位置建议保持默认位置,并保存位置图片
安装结果
(之前版本安装还得手动设置环境变量,现在不用了,直接测试:nvcc -V)
4.安装cuDNN
解压下载好的压缩包,得到cudnn-windows-x86_64-8.9.5.29_cuda12-archive文件夹:
将cudnn文件中对应的文件移动到cuda对用安装目录中,注意不要移动到错误的位置
测试一下CUDA和cudnn是否安装成功
进入目录
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.0\extras\demo_suite
执行上面两个程序会出现以下结果,表示成功
以上就是CUDA部署过程,欢迎批评指正,感谢点赞加关注!