一,如果你不是在ros中使用的:
只需要修改darknet文件下的Makefile文件的三处位置
第一处:更改Makefile前两行GPU和CUDNN的配置:
GPU=1
CUDNN=1
第二处:更改CUDA的路径(不一的要改,可以先确认一下你的路径是不是正确的)
48~51行,在"ifeq ($(GPU), 1)"语句块中修改为自己的CUDA安装路径,更改前默认路径如下:
ifeq ($(GPU), 1)
COMMON+= -DGPU -I/usr/local/cuda/include/
CFLAGS+= -DGPU
LDFLAGS+= -L/usr/local/cuda/lib64 -lcuda -lcudart -lcublas -lcurand
例如修改为:
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(要改成自己的位置)
第三处:23行,修改NVCC的路径(不一定要改,先看看路径和你的是否一样,不一样再改):
NVCC=/usr/local/cuda-8.0/bin/nvcc(要改成自己的位置)
配置完成后保存退出,输入make编译以下就好了。
二,如果是在ros中的
只需要修改darknet_ros/darknet文件下的Makefile文件的三处位置
第一处:更改Makefile前两行GPU和CUDNN的配置:
GPU=1
CUDNN=1
第二处:更改CUDA的路径(不一的要改,可以先确认一下你的路径是不是正确的)
48~51行,在"ifeq ($(GPU), 1)"语句块中修改为自己的CUDA安装路径,更改前默认路径如下:
ifeq ($(GPU), 1)
COMMON+= -DGPU -I/usr/local/cuda/include/
CFLAGS+= -DGPU
LDFLAGS+= -L/usr/local/cuda/lib64 -lcuda -lcudart -lcublas -lcurand
例如修改为:
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(要改成自己的位置)
第三处:23行,修改NVCC的路径(不一定要改,先看看路径和你的是否一样,不一样再改):
NVCC=/usr/local/cuda-8.0/bin/nvcc(要改成自己的位置)
配置完成后保存退出,输入make编译,
编译完成后到编译工作空间:catkin_make