WIN10 系统下下载安装YOLOV4并使用GPU加速(保姆级)

本文详述了在Windows 10系统中,如何下载安装YOLOv4并利用GPU加速的全过程,包括下载源码、配置opencv、安装CUDA和CUDNN、设置Visual Studio以及解决配置过程中遇到的问题。在安装过程中,重点强调了环境变量的设置和CUDA配置的注意事项,以确保YOLOv4能成功运行并利用GPU提升运行效率。
摘要由CSDN通过智能技术生成

这次因为比赛需要测试一下yoloV4的速度,就安装了一下YOLOV4,差点哭出来 基本上把能踩的坑都踩了一遍(无奈)。装完之后想写一片总结记一下,也帮大家伙避避雷。

一、下载yoloV4源码

建议使用git下载,一行指令就搞定还挺快的。(没有的话可以直接在github上点下载就好,不过这个我没试过)
在需要下载的位置右键点击Git Bash Here(我是在E盘下)

在这里插入图片描述
打开git界面后输入:

git clone https://github.com/AlexeyAB/darknet.git

等待下载即可
下载后当前目录下会出现一个名为darknet的文件夹,这个就是我们需要的yoloV4的源码

二、opencv的下载和配置

我用的是opencv4.4。opencv在Windows下还是很好配置的。
从官网(https://opencv.org/)点击下载就行yoloV4要求opencv的版本在2.4以上,所以opencv3或者opencv4都是可以的。下载完成后会有下图所示的exe文件直接双击,跟随引导解压即可。
在这里插入图片描述
解压完成后要将其添加到环境变量中去:
桌面右键单击此电脑->属性->点击左侧的"高级系统设置"
会出现下图界面:
在这里插入图片描述
点击环境变量,在用户变量的PATH中填入路径(以opencv4.4.0为例):
你的安装地址\opencv\build\x64\vc15\bin
在系统变量中添加:
①变量名:OPENCV_DIR,路径:你的安装地址\opencv\build

使用GPU加速训练Yolov8,你需要确保以下几个步骤已经完成: 1. 安装GPU驱动程序:确保你的计算机已经安装了适当的GPU驱动程序。你可以从GPU供应商(如NVIDIA)的官方网站上下载并安装最新的驱动程序。 2. 安装CUDACUDA是NVIDIA提供的用于GPU加速的并行计算平台和编程模型。你需要根据你的GPU型号下载并安装适合的CUDA版本。确保在安装CUDA时按照官方文档的指导进行操作。 3. 安装cuDNN:cuDNN是NVIDIA提供的用于深度神经网络加速的库。你需要从NVIDIA开发者网站上下载适合的cuDNN版本,并按照官方文档的说明进行安装。 4. 设置深度学习框架:根据你选择的深度学习框架,需要进行相应的设置以支持GPU加速。以PyTorch为例,可以通过设置`torch.cuda.is_available()`来检查是否可用GPU。 5. 加载数据和模型:在训练Yolov8之前,确保你已经准备好了训练数据集和相应的模型架构。可以使用GPU进行数据加载和模型初始化。 6. 设置运行参数:在训练脚本中,你可以通过设置相关的参数来启用GPU加速。例如,在PyTorch中,可以使用`model.cuda()`将模型移至GPU上,使用`torch.nn.DataParallel`进行多GPU训练等。 7. 启动训练过程:最后,你可以执行训练脚本来开始使用GPU加速训练Yolov8。在训练过程中,你应该可以看到GPU使用率明显增加。 通过以上步骤,你应该能够成功地使用GPU加速训练Yolov8。请注意,具体的步骤可能因你的操作系统GPU型号和深度学习框架而有所不同,请参考相关文档和教程以获取更详细的指导。
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值