系统:ubuntu 自带cuda10.0
1、下载与安装darknet
git clone https://github.com/AlexeyAB/darknet
cd darknet
2、以下步骤我都在直接进入commit中操作
(1)核对cuda,若输入nvcc -V 出现命令位找到,则说明未配置cuda,继续阅读,否则直接执行第3步
nvcc -V
-bash: nvcc: command not found
(2)将cuda加入路径
sudo chmod -R 777 ~/.bashrc
gedit ~/.bashrc
(3)在文件最后加入以下三行
export CUBA_HOME=/usr/local/cuda-10.2
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-10.2/bin:$PATH
(4)执行使cuda生效
source ~/.bashrc
(5)可以输入nvcc -V查看版本
3、修改Makefile文件
cd darknet
gedit Makefile
进入Makefile,修改
GPU=1
CUDNN=1
OPENCV=1 # 如果安装了opencv则改为1
OPENMP=0
DEBUG=0
...
NVCC=/usr/local/cuda-10.0/bin/nvcc # 原为 NVCC = nvcc,cuda后面的紧跟自己的cuda版本
4、输入make -j2
5、在darknet文件夹中查看是否有yolov4-t

本文介绍了如何在Ubuntu系统和Jetson Nano设备上部署YOLOv3、YOLOv4以及它们的tiny版本。首先,确认CUDA环境并将其添加到路径中,接着修改Makefile文件,然后编译darknet。最后,检查权重文件和配置文件,并展示了视频检测的使用方法。
最低0.47元/天 解锁文章
466

被折叠的 条评论
为什么被折叠?



