文章目录

前言

在训练 darknet yolo 时碰到这个问题,这是由于新旧版本的 cudann 中的文件有区别,新版本多了一些文件。
在这里插入图片描述

解决方法

  • 回到之前安装 cudnn 的那篇博客,找到了自己当初安装的 cudnn 的版本。
  • 进入官网:https://developer.nvidia.com/rdp/cudnn-archive,下载自己之前安装的版本(需要登陆)。
    在这里插入图片描述
  • 解压后,进入文件夹,执行:
    # 参考来源:https://github.com/pjreddie/darknet/issues/2356
    sudo cp cuda/include/cudnn*.h /usr/local/cuda/include # 新版的使用这条命令可以将cudnn*.h多个文件复制到指定文件夹
    sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64  
    sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* 
    

在这里插入图片描述

Logo

瓜分20万奖金 获得内推名额 丰厚实物奖励 易参与易上手

更多推荐