首先现在CUDA和CUDNN并安装,我下载的是10.1版本,这里分享百度网盘链接:
链接:https://pan.baidu.com/s/1BhzlC9WUmHlyOLHsFlaKLw
提取码:nk89
有两个版本的yolo
第一个是基于Keras/TensorFlow框架的YOLOv3模型训练方法,该方法操作起来比较便捷,也容易上手,但是用CPU训练模型
第二次是Darknet框架使用GPU训练模型,仅需要66个小时,速度快得让人睡着都能笑醒。由于大多数教程都是针对Linux系统,所以看了很多博文,都无法正确完成配置,所以本文将详细讲解Windows系统的darknet框架配置。我训练6000多张图片大约要3天左右的时间
电脑环境:
Windows 10 64位(查看方式:此电脑——右击选择属性)
NVIDIA GeForce GTX 1660 (查看方式:控制面板——设备管理器——显示适配器)
GTX1660 支持CUDA的最大版本:CUDA 10.1
很多博客说需要下载与 GT 730 支持的版本一直的CUDA和cuDNN,即10.1,然而我尝试了2次均会出错,所以最终的搭配为CUDA10.1+cuDNN7.4+ tensorflow-gpu2.0 + python3.6+Visual Studio 15+opencv3.4.0
一、文件下载准备
(1)下载安装CUDA 10.1和cuDNN 7.4
CUDA及CUDNN的安装请参考下面大佬博文,安装时不要轻易改