【亲测】centos 7 下安装cuDNN

本文档记录了在CentOS 7系统中安装cuDNN的详细步骤,包括从官方文档获取参考资料,准备安装文件,解压缩并复制到CUDA指定目录,设置文件权限,以及验证cuDNN安装成功的全过程。
摘要由CSDN通过智能技术生成

【亲测】centos 7 下安装cuDNN

【参考】

  1. http://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html

【文件准备】

  1. cuda_8.0.61_375.26_linux.run
  2. cudnn-8.0-linux-x64-v5.1.tgz

【步骤】

  1. 切换到压缩包所在位置,解压
    $ tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz

  2. 复制相关文件到cuda特定目录下(我的cuda安装目录为/usr/local/cuda/)

    $ sudo cp cuda/include/cudnn.h /usr/local/cuda/include
    $ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

  3. 修改文件权限
    $ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

  4. 使用
    这个时候编译caffe可以将Makefile.config中的USE_CUDNN不注释,从而使用cuDNN加速
    # cuDNN acceleration switch (uncomment to build with cuDNN).
    USE_CUDNN := 1

【附录】

<
  • 3
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值