yolov4训练数据: cuda和cudnn的安装
创始人
2024-02-28 17:15:23

近期再搞openvino+Yolov4目标检测,记录一下cuda和cudnn的安装笔记。

mirrors / alexeyab / darknet · GitCode

1.cuda的安装

cuDNN Archive | NVIDIA Developer

wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
sudo sh cuda_10.2.89_440.33.01_linux.run

执行上面的命令即可安装成功;

允许nvcc -V可以查看是否安装成功;

之前我的版本是cuda11.8

至于为什么降低版本?

因为在darknet train训练数据出现如下错误:

 这里出现CUDA Error: no kernel image is available for execution on the device;

排查各种原因后,进行各种尝试后还是报此错,最终降低版本解决此问题;

第一步排查:显卡对应的算力是否对应(网上大多数都是解释是算力与显卡不匹配造成的)

训练显示可知  GTX 950M对应的算力是50,因此darknet的配置文件就选对应的ARCH参数:

也可以通过官网查询:CUDA GPUs - Compute Capability | NVIDIA Developer

 第二步排查cuda的版本对应的算力

 可知cuda11.8是支持5.0的。

第三步是修改cfg配置文件参数:

 修改对应的subdivisions的参数值,但注意subdicisions小于等于batch数值;

查看显卡信息:

发现经过无论怎么尝试都会报错,按道理不应该是版本问题,最终降低了版本就可以了,可能与显卡不匹配吧。

为什么一开始下载的是cuda11.8,就是根据显卡信息cuda version 信息下载的。

所以版本的选择还是多根据自己的显卡进行选择,免得需要重新安装,并不是一味的认为最高的就是最好的。这里11.8应该是指的是最高版本。

这里有个现象:local下面会有两个cuda的文件夹(最开始还以为是我的路径有问题)

 ls -l后 才发现是链接出来的,很是尴尬。

2.cudnn的安装

cuDNN Archive | NVIDIA Developer

根据对应的cuda版本下载对应的版本号,注意此处需要注册账号才可以下载,如下:

安装命令:

cd cudnn-linux-x86_64-8.6.0.163_cuda10-archive/
sudo cp include/cudnn* /usr/local/cuda/include
sudo cp lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*
sudo chmod a+r /usr/local/cuda/lib64/cudnn*

复制到对应的cuda文件夹下即可。

如果不匹配darknet训练的时候会报错:cudnn的库文件错误之类的ERROR。

测试:

①时间

最开始是no GPU的方式训练数据,时间基本上几十小时以上,现在用了GPU,时间节省了大部分时间只要几小时而已,具体根据图像的数量和配置参数来定。

②准确率

GPU训练:

 no GPU训练:

 

同样的训练数据在不同情况下进行训练后,相同的图片的测试结果:

NO GPU: 52% 69% 61%     451ms/248ms/255ms

GPU      :100% 100% 100% 31ms/32ms/32ms

因此,无论是学习还是开发,建议还是都是使用GPU进行数据训练,有无GPU的差别很大. 

参考:

CUDAError:nokernelimageisavailableforexecutiononthedevice-爱码网英伟达显卡不同CUDA支持的计算能力情况及不同算力对应显卡列表_Zency_SUN的博客-CSDN博客_英伟达cuda

linux 卸载cuda_如何卸载CUDA - 全栈程序员必看

Linux安装CUDA_C_oarnge的博客-CSDN博客_linux安装cuda

相关内容

热门资讯

埃菲尔铁塔在哪 中国仿建埃菲尔... 2019年4月26日,广西南宁市,街头惊现一座巨型山寨版埃菲尔铁塔,高约20米,白色塔身,造型逼真,...
北京的名胜古迹 北京最著名的景... 北京从元代开始,逐渐走上帝国首都的道路,先是成为大辽朝五大首都之一的南京城,随着金灭辽,金代从海陵王...
苗族的传统节日 贵州苗族节日有... 【岜沙苗族芦笙节】岜沙,苗语叫“分送”,距从江县城7.5公里,是世界上最崇拜树木并以树为神的枪手部落...
应用未安装解决办法 平板应用未... ---IT小技术,每天Get一个小技能!一、前言描述苹果IPad2居然不能安装怎么办?与此IPad不...
脚上的穴位图 脚面经络图对应的... 人体穴位作用图解大全更清晰直观的标注了各个人体穴位的作用,包括头部穴位图、胸部穴位图、背部穴位图、胳...
长白山自助游攻略 吉林长白山游... 昨天介绍了西坡的景点详细请看链接:一个人的旅行,据说能看到长白山天池全凭运气,您的运气如何?今日介绍...
demo什么意思 demo版本... 618快到了,各位的小金库大概也在准备开闸放水了吧。没有小金库的,也该向老婆撒娇卖萌服个软了,一切只...
猫咪吃了塑料袋怎么办 猫咪误食... 你知道吗?塑料袋放久了会长猫哦!要说猫咪对塑料袋的喜爱程度完完全全可以媲美纸箱家里只要一有塑料袋的响...
世界上最漂亮的人 世界上最漂亮... 此前在某网上,选出了全球265万颜值姣好的女性。从这些数量庞大的女性群体中,人们投票选出了心目中最美...
埃菲尔铁塔在哪 中国仿建埃菲尔... 2019年4月26日,广西南宁市,街头惊现一座巨型山寨版埃菲尔铁塔,高约20米,白色塔身,造型逼真,...
苗族的传统节日 贵州苗族节日有... 【岜沙苗族芦笙节】岜沙,苗语叫“分送”,距从江县城7.5公里,是世界上最崇拜树木并以树为神的枪手部落...
北京的名胜古迹 北京最著名的景... 北京从元代开始,逐渐走上帝国首都的道路,先是成为大辽朝五大首都之一的南京城,随着金灭辽,金代从海陵王...
应用未安装解决办法 平板应用未... ---IT小技术,每天Get一个小技能!一、前言描述苹果IPad2居然不能安装怎么办?与此IPad不...