darknet yolo 使用gpu

darknet YOLO 使用GPU训练

修改Makefile文件，如下：

GPU=1
CUDNN=1
OPENCV=1
OPENMP=0
DEBUG=0

flag置1，表示使用opencv和cuda

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

这里的路径改成本机上安装cuda的路径

NVCC也修改，否则会报错nvcc:not found

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

make clean 在进行make，即可使用gpu。