ubuntu darknet GPU版

darkNet GPU版修改测试

cuda、cudnn、opencv 安装

1. 版本信息查看

cuda版本信息查看

cat /usr/local/cuda/version.txt
 显示
 
CUDA Version 10.0.130

cudnn 版本信息查看

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

显示
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 3
#define CUDNN_PATCHLEVEL 1
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#include "driver_types.h"

opencv 版本查看

pkg-config --modversion opencv

3. Darknet makefile 文件配置

GPU=1
CUDNN=1
OPENCV=1


NVCC=/usr/local/cuda-10.0/bin/nvcc 

ifeq ($(GPU), 1) 
COMMON+= -DGPU -I/usr/local/cuda-10.0/include
CFLAGS+= -DGPU
LDFLAGS+= -L/usr/local/cuda-10.0/lib64 -lcuda -lcudart -lcublas -lcurand
endi