本文详细介绍了如何在Windows系统下，利用GPU训练YOLOv4模型。包括安装CUDA和CUDNN的步骤，配置CUDA环境变量，检查CUDA安装是否成功。此外，还讲解了opencv3.4的下载和配置，以及YOLOv3的darknet项目编译过程，涉及到Visual Studio的设置和dll文件的处理。最后，通过命令行展示了如何测试模型效果。
在windows下训练yolov4模型（GPU）
1.为电脑安装最新版本的cuda和cudnn
cuda安装地址：https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
cudnn安装地址：https://developer.nvidia.com/cudnn
其中cudnn下载安装包需要注册一个账号，过程是免费的。
两个文件下载完成之后是这样的：
1，将cuda运行安装，建议默认路径，因为我试过自己修改路径，最后遇到了两个问题：
- ①临时解压路径：
- ②安装路径：
- （应该是CUDA10，因为我已经安装完了，所以在网上找了一个图来表示说明）
所以会遇到这两个路径如果设置为同一个的话，安装时会自动删除那个解压缩的Temp所在的文件夹，最后会发现CUDA安装的文件又没了！所以建议默认路径原因就是怕这两个路径设置为同一个了。
2，对于cudnn直接将其解开压缩包，然后需要将以下bin,include,lib文件复制粘贴到cuda的文件夹下C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1：
最低0.47元/天 解锁文章
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
