Centos 安装Caffe

白水baishui

于 2021-09-15 08:52:45 发布

阅读量459

点赞数 1

CC 4.0 BY-SA版权

分类专栏： # 深度学习 # Linux 文章标签： caffe centos 安装错误 linux

本文链接：https://blog.csdn.net/baishuiniyaonulia/article/details/120300951

深度学习同时被 2 个专栏收录

16 篇文章

订阅专栏

Linux

15 篇文章

订阅专栏

本文档详细介绍了CUDA和CUDNN的下载及安装步骤，特别是针对CUDA 10.2和CUDNN v7的版本匹配。首先从官方网站下载对应版本，然后解压并移动cudnn的头文件和库文件到CUDA安装目录。通过检查cudnn版本确认安装成功。接着，提供了Caffe的安装链接，帮助读者在CentOS 7上进行安装配置。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. CUDA和CUDNN

cuda下载地址：https://developer.nvidia.com/cuda-toolkit-archive
cudnn下载地址：https://developer.nvidia.com/rdp/cudnn-archive

注意当前版本caffe只支持 cuda8.0-10.2 和 cudnn v7 版本，对照自己的cuda版本下载对应的cudnn

cuda的安装过程网上都有我就不详述了，讲一下cudnn如何安装，下载好cudnn后，解压出来：

tar -xzvf cudnn-10.2-linux-x64-v7.6.5.32.tgz

如何把cudnn的文件移到cuda中：

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

完毕后执行以下命令查看版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

显示CUDNN_MAJOR 7则说明cudnn安装成功：

#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 5
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#include "driver_types.h"