未发现CUDNN正在运行YOLO DARKNET

Question

我正在使用此命令在终端中运行用于对象检测训练的darknt

./darknet detector train cfg/voc.data cfg/yolov3-voc.cfg darknet53.conv.74

但是，我有一个这样的错误：

./darknet: error while loading shared libraries: libcudnn.so.8: cannot open shared object file: No such file or directory

我该如何解决这个问题？

Answer 1

我希望我可以发表评论，但我必须使用答案来回复(没有足够的rep(。

如果在Makefile中设置GPU=1，PJ Reddie的Darknet将使用CuDNN。CuDNN是nvidia GPU特有的。与中一样，如果您没有GPU或没有安装Cuda Toolkit(并且您有GPU(，它将不起作用。如果我告诉你一些你已经知道的事情，我会认为是CUDA Toolkit安装出了问题。如果这没有帮助，我很抱歉。

如果你有GPU，但它仍然不工作，我会尝试重新安装它。查看Redmon先生网站上的安装指南，了解更多详细信息：https://pjreddie.com/darknet/install/

Answer 2

我对此的回应可能有点太晚了，但对于任何有此错误的人来说：

Error: You set incorrect value batch=1 for Training! You should set batch=64 subdivision=64.

问题出现在cfg/yolov3-voc.cfg文件中。在第一行中，您必须编辑批次大小。