【Ubuntu|cuDNN】训练 DarkNet 时解决 fatal error: cudnn_version.h
文章目录前言解决方法前言在训练 darknet yolo 时碰到这个问题,这是由于新旧版本的 cudann 中的文件有区别,新版本多了一些文件。解决方法回到之前安装 cudnn 的那篇博客,找到了自己当初安装的 cudnn 的版本。进入官网:https://developer.nvidia.com/rdp/cudnn-archive,下载自己之前安装的版本(需要登陆)。解压后,进入文件夹,执行:#
·
前言
在训练 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*
更多推荐
已为社区贡献3条内容
所有评论(0)