yolov4的GPU搭建配置

下载yolov4的框架
https://github.com/AlexeyAB/darknet

解压darknet-master.zip
打开darknet-master文件夹

在CMaketLists.txt中找到if(ENABLE_OPENCV)
添加  set(OpenCV_DIR E:/cuda_opencv/opencv/build/install)

打开powershell控制台
输入   .\build.ps1
可能会出现没有权限, 跟打开cmd一样在外部打开,解锁权限

训练自己的数据集,参考

https://mp.csdn.net/console/editor/html/107311683

生成dll与lib文件
打开H:\darknet-master\build\darknet文件,如果是GPU版本,点击yolo_cpp_dll.sln,改为Release X64运行
打开yolo_cpp_dll.vcxproj,如果cuda版本不是10.0,需要修改

配置yolo_cpp_dll.sln
项目 --> 属性 --> 配置属性 --> C/C++ --> 预处理器 --> 取消预处理器定义,查看所需预处理器是否被禁用

项目 --> 属性 --> 配置属性 -->CUDA C/C++ --> Device --> Coda Generation
将compute_30, sm_30,compute_75, sm_75改为自己的GPU运算能力
compute_61, sm_61(GTX 1080Ti)
通过ht

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
为了在YOLOv8中配置GPU环境,你需要遵循以下步骤: 1. 确保你的电脑已经安装了适当的显卡驱动。你可以通过访问NVIDIA官方网站来找到适合你显卡的最新驱动程序。 2. 安装CUDA工具包。CUDA是一种用于并行计算的平台和编程模型,可以充分利用GPU的计算能力。你可以从NVIDIA开发者网站下载适合你系统的CUDA版本,并按照指示进行安装。 3. 安装cuDNN库。cuDNN是一个用于深度神经网络的GPU加速库,可以提高YOLOv8的训练和推理速度。你需要从NVIDIA开发者网站下载适合你CUDA版本的cuDNN库,并按照指示进行安装。 4. 配置PyTorch的GPU环境。确保你已经正确安装了PyTorch,并根据你的系统和CUDA版本选择适合的PyTorch版本。在代码中,你需要使用`torch.cuda.is_available()`来检查GPU是否可用,以及使用`torch.cuda.device()`来选择使用的GPU设备。 5. 下载YOLOv8的代码和预训练权重。你可以从YOLOv8的GitHub仓库中获取代码,并下载相应的预训练权重文件。 6. 运行YOLOv8。使用配置好的GPU环境,你可以运行YOLOv8的训练或推理代码。确保在代码中正确设置了GPU设备,并加载了预训练权重。 请注意,以上步骤仅提供了一般的指导,具体的配置步骤可能因你的操作系统、GPU型号和软件版本而有所不同。建议你在进行配置之前参考引用的资料、和,以获得更详细和准确的配置信息。 #### 引用[.reference_title] - *1* *2* [YoloV4GPU环境配置](https://blog.csdn.net/straker/article/details/126924600)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] - *3* [yolov5 GPU运行环境搭建(不详细版本)](https://blog.csdn.net/m0_67313306/article/details/128179039)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值