CUDA配套cuDNN的安装

安装CUDA10配套cuDNN,给需要的人参考下。
参考:官网原版安装指南
作者:Su Liang,2018-10-19
cuDNN是nvidia开发的deep neural network library(dnn),即为深度神经网络GPU加速库。
提供了前向/后向/卷积/池化/标准化/激活等层。

(1)注册

https://developer.nvidia.com/rdp/form/cudnn-download-survey

(2)选择cuDNN版本:需要跟CUDA的版本匹配,由于安装的是CUDA10,需选择cuDNN v7.3.1

这里选择了cuDNN v7.3.1 library for linux。没有选择runtime library forubuntu 16.04,安装说明里说,cuDNN v7.3.1 lib for linux这种tar文件(下载的是tgz压缩文件)适用于所有linux系统,而debian文件适用于ubuntu系统。

(3)安装前提:

  • GPU
  • x86_64 – ubuntu 14.04 or ubuntu 16.04
  • CUDA以及nvidia驱动

(4)进入下载目录,解压缩cuDNN安装文件

tar -xzvf cudnn-9.0-linux-x64-v7.tgz

(5)拷贝如下解压缩文件到CUDA toolkit的目录中,然后变更文件允许

不要遗漏命令最后那个*

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 /usr/local/cuda/lib64/libcudnn*

即完成安装了。

(6)查证是否安装成功

nvcc -V
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值