deep-learning - 未找到运行 YOLO DARKNET 的 CUDNN

Question

我正在使用此命令在终端中运行 darkent 进行对象检测训练

./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

我该如何解决这个问题？

标签： deep-learningobject-detectionyolodarknet

Answer 1

我希望我可以发表评论，但我将不得不使用答案来回复（没有足够的代表）。

GPU=1如果您在 Makefile 中设置，PJ Reddie 的 Darknet 使用 CuDNN 。CuDNN 特定于 nvidia GPU。例如，如果您没有 GPU 或没有安装 Cuda Toolkit（并且您确实有 GPU），它将无法工作。如果我告诉你一些你已经知道的事情，我会假设是 CUDA Toolkit 安装出错了。对不起，如果这没有帮助。

如果您有 GPU 并且仍然无法正常工作，我会尝试重新安装它。更多详情请查看 Redmon 先生网站上的安装指南：https ://pjreddie.com/darknet/install/