在谷歌 colab 上组合 Makefile YOLOV4 时出错 Cuda？

Question

enter image description here

我在 google colab 上训练 yolov4 时遇到了“include/darknet.h:664:5: error: ‘cudnnTensorDescriptor_t’ does not name a type”错误。

Answer 1

尝试添加这些行：https://colab.research.google.com/drive/1Ej84fB-pMzcxgTEPBqr4y790v3wH9UfE?usp=sharing

请不要忘记根据您的 GPU 更改 compute_capability

然后，选择您喜欢的以下设置：

GPU=1
CUDNN=1
CUDNN_HALF=0
OPENCV=1
AVX=0
OPENMP=0
LIBSO=1
ZED_CAMERA=0
ZED_CAMERA_v2_8=0

你可以参考这个来探索你的选择：https://github.com/AlexeyAB/darknet#how-to-compile-on-linux-using-make

colab 行的参考： https://colab.research.google.com/drive/1mzL6WyY9BRx4xX476eQdhKDnd_eixBlG#scrollTo=GNVU7eu9CQj3