就不啰嗦其它的了,直接开始,本教程以Windows电脑为例。
1、安装CUDA
操作前建议先更新你电脑英伟达显卡驱动
网上很多教程说在安装前,需要查看本电脑的cuda版本,电脑是哪个版本就只能安装哪个版本,其实这部不是必须的,不需要看自己电脑是什么版本就必须安装哪个版本。
打开cuda toolkit下载列表链接
https://developer.nvidia.com/cuda-toolkit-archive
你需要哪个版本就点击选择哪个版本,
根据你的实际情况进行选择,如果你的下载速度比较慢的话可以鼠标右键点击download按钮,选择复制链接地址,然后将链接复制到迅雷里进行下载,下载速度非常快,我的都能达到100多M,
将exe安装包下载到电脑上之后双击运行安装,安装过程全程保持默认安装就可以了。
2、安装CUDNN
https://developer.nvidia.com/rdp/cudnn-archive,打开链接,你的是什么版本就下载什么版本就可以了
还是用迅雷下载:
鼠标右键点击下载链接,选择复制链接地址,添加到迅雷里,
迅雷里添加下载链接的时候要删除最后的的“/”
保证下载的是ZIP类型文件
将CUDNN压缩包下载完成之后解压出来,复制里面的三个文件夹
将这三个文件夹复制到CUDA目录内,
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4
因为我安装的是12.4版本,所以文件夹目录最后是v12.4,你根据自己实际情况找到你的CUDA版本文件夹,CUDNN三个文件夹复制进来之后就安装完成了