YOLOv11 GPU环境搭建与问题分析
安装CUDA
CUDA(Compute Unified Device Architecture)是一种由NVIDIA开发的并行计算平台和编程模型,主要用于利用GPU(图形处理单元)的强大并行计算能力来加速各种复杂的计算任务。
一、查看电脑支持的版本
- 查看当前你的电脑显卡支持的最高的CUDA版本。命令:nvidia-smi
方法二:
【鼠标右键桌面】--【显示更多选项】--【NVIDIA控制面板】--在控制面板的左下角点击【系统信息】--选择【组件】
二、下载CUDA
- CUDA官网地址:CUDA Toolkit Archive | NVIDIA Developer
【注意】选择自己的系统和版本。如果网络稳定可以选exe(network),如果不稳定建议选前者。
三、安装CUDA
1、双击下载的exe安装文件进行安装。
- 下面的路径是提取地址,安装完后会自动删除,选择默认即可,点击【OK】
2、下一个路径用默认或者自定义都可。
【注意】不要与上一个临时目录相同,否则会被删除。
3、“安装选项”建议选“自定义C(高级)”
四、安装结果确认
cmd命令行下,使用nvcc -V命令,查看CUDA是否安装成功,如果出现如下界面表示CUDA安装成功。
安装cuDNN
是由NVIDIA开发的一个深度学习GPU加速库,专门用于加速深度神经网络(DNN)的训练和推断过程。
一、下载cuDNN
下载地址: