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次均会出错&#