Windows10+Darknet+YOLOv3训练自己数据(GPU)——2.CUDA+cuDNN+Darknet框架配置

本文详细介绍了在Windows10系统下,如何配置CUDA 9.0、cuDNN 7.1和Darknet框架,以利用GPU加速YOLOv3模型的训练。通过解决各种配置问题,最终成功生成darknet.exe文件。
摘要由CSDN通过智能技术生成

2.CUDA+cuDNN+Darknet框架配置

阅读本文前请先阅读:

Windows10+Darknet+YOLOv3训练自己数据——1.数据集准备

第一次尝试了博主基于Keras/TensorFlow框架的YOLOv3模型训练方法,该方法操作起来比较便捷,也容易上手,但是用CPU训练模型,我有2000张图片,训练时粗略计算了一下需要750+h的时间,整整31个昼夜啊!!!万一最后模型不理想,那岂不是要搞好几个月。所以后面果断放弃,如果训练样本少的小伙伴可以尝试此法,的确很便捷。

第二次基于Darknet框架使用GPU训练模型,仅需要66个小时,速度快得让人睡着都能笑醒。由于大多数教程都是针对Linux系统,所以看了很多博文,都无法正确完成配置,所以本文将详细讲解Windows系统的darknet框架配置。

电脑环境:
Windows 10 64位(查看方式:此电脑——右击选择属性)
NVIDIA GeForce GT 730 (查看方式:控制面板——设备管理器——显示适配器)
GT 730 支持CUDA的最大版本:CUDA 10.1

很多博客说需要下载与 GT 730 支持的版本一直的CUDA和cuDNN,即10.1,然而我尝试了2次均会出错&#

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值