前言
CUDA是 NVIDIA 开发的并行计算平台和编程模型,让开发者能用 NVIDIA 显卡(GPU)进行通用计算。
GPU有成百上千个计算核心,擅长并行处理大量简单计算任务,而CUDA让你用类似C语言的方式编写程序,调用GPU加速计算。
它广泛用于深度学习、科学计算、图形渲染等领域,可以把原本需要几天的训练任务加速到几小时甚至几分钟。
确定需要下载的Cuda版本
需要先下载英伟达显卡驱动,这里略过。
右键开始图标,点击:终端管理员
输入命令
nvidia-smi
可知,需要的版本为:13.2
下载安装包
官方下载链接:https://developer.nvidia.com/cuda/toolkit
安装
右键安装包,以管理员的方式运行
等待
保持默认,点击:下一步
保持默认,点击:下一步
同意协议,点击:下一步
等待安装结束
验证
右键开始图标,点击:终端管理员
输入命令
nvcc -V
安装cuDNN
下载安装包
官方下载链接:https://developer.nvidia.com/cudnn
右键安装包,以管理员的方式运行
保持默认,点击:下一步。直至结束
至此结束全部安装。