系统: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=