安装darknet
- 下载yolov4:
git clone https://github.com/AlexeyAB/darknet.git
- 编译,如果用CPU直接编译即可,如果用GPU跑,需要设置CUDA=1,CUDNN=1
$cd darknet
$make
- 编译报错:
nvcc fatal : Unsupported gpu architecture 'compute_30'
Makefile:177: recipe for target 'obj/convolutional_kernels.o' failed
make: *** [obj/convolutional_kernels.o] Error 1
- 这是由于计算机显卡型号和算力需要在makefile里进行匹配,可以查阅makefile对应修改的型号,这里我是RTX2060,在makefile里没有写,我对应的是computer_61,也就是把
-gencode arch=compute_61,code=[sm_61,compute_61]
之前的都删去。
# Tesla A100 (GA100), DGX-A100, RTX 3080
# ARCH= -gencode arch=compute_80,code=[sm_80,compute_80]
# Tesla V100
# ARCH= -gencode arch=compute_70,code=[sm_70,compute_70]