第一部分:CUDA的基本作用
是一个并行的计算架构,可以在显卡上执行计算任务,可以通过将复杂的计算任务分配到多个处理机上进行并行的处理来提高计算速度,它主要是用在科学计算,人工智能和游戏开发领域。
第二部分:CUDA的下载安装
(1)CUDA的安装条件
首先你的电脑上需要有显卡,其次,你的显卡还不能是AMD的
查看自己显卡的方式:
同时按住ctrl+shift+point数字键,打开任务管理器:
然后点击性能:
上面就可以看到我的显卡GPU是3060版本。
(2)确定自己需要下载的CUDA版本
这一点尤其需要注意,CUDA不能随意进行下载安装的,要下载适合于自己的版本!
进入电脑桌面,单击右键:
如果右键找不到NVIDIA控制面板,看我的博客:电脑桌面右键“没有NVIDIA控制面板”的解决方法-CSDN博客
然后点击系统信息:
记住自己的驱动程序版本号:
(3)查找对应CUDA版本并下载
核实cuda版本网址:CUDA 12.6 Update 2 Release Notes
我的是556.12版本,我选一个能够0满足条件的即可:
如果上述的这些版本查找感觉比较麻烦,也可以用命令行的操作:
nvidia-smi
下载cuda网站:CUDA Toolkit Archive | NVIDIA Developer
点击之后进入详细下载界面:
然后点击右下角的下载:
(4)安装CUDA
通过网盘分享的文件:cuda_12.5.0_555.85_windows.zip
链接: https://pan.baidu.com/s/1NPjNhZeZR-e9lWg30G02Eg?pwd=q6we 提取码: q6we
--来自百度网盘超级会员v5的分享
上面是刚刚下载得到的cuda安装包,下面我们开始进行安装cuda.
不用修改路径。
点击同意。
点击下一步。
正常情况下是安装没有问题的,但是如果安装失败,出现“NVDIA安装程序失败”的问题,可以看下我的博客:cuda安装出现“NVDIA安装程序失败”问题解决方法-CSDN博客
(5)检查是否安装成功
输入指令:
nvcc -V
第三部分:CUDNN下载安装
(1)下载链接
cuDNN Archive | NVIDIA Developer
我下载好的:
通过网盘分享的文件:cudnn-windows-x86_64-9.2.0.82_cuda12-archive.zip
链接: https://pan.baidu.com/s/1RpreJdTXhgcPa_hiTuNXsA?pwd=5kjy 提取码: 5kjy
--来自百度网盘超级会员v5的分享
(2)应用
查看cuda的安装位置:
set cuda
上述的相同文件夹的内容会自动合并。
第四部分:额外信息
(1)查看cuda版本
import torch
print("PyTorch Version:", torch.__version__)
print("CUDA Available:", torch.cuda.is_available())
print("CUDA Version:", torch.version.cuda)
print("Device Count:", torch.cuda.device_count())
(2)查看cuda位置
set cuda
(3)查看cudnn版本
import torch
torch.backends.cudnn.is_available()#查询cudnn是否可用
print(torch.backends.cudnn.version())