yolo目标检测windows下GPU版本配置和模型训练(一)

本文详细介绍了在Windows 10环境下,如何配置CUDA 10.1、cuDNN 7.4,以及安装Visual Studio 2015、opencv3.4和Darknet,解决CUDA版本匹配问题,最终实现GPU加速的YOLOv3模型训练。通过步骤解析,帮助读者在遇到配置问题时找到解决方案。
摘要由CSDN通过智能技术生成

首先现在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的安装请参考下面大佬博文,安装时不要轻易改

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值