ubuntu中安装cuDNN

几经周折后终于安装好了cuda,接下来根据tensorflow的要求,开始安装cudnn
官方教程地址

1. 下载cudnn

要注册一个nvidia的帐号才可以下载
下载地址
根据系统版本,选择相应的版本。我的是ubuntu,选择通过压缩包的形式安装
cudnn-9.0-linux-x64-v7.tgz

2. 安装

然后解压

tar -xvzf cudnn-9.1-linux-x64-v7.1.tgz 

会在当前目录下创建一个 cuda 的文件夹,包括以下文件。

cuda/include/cudnn.h
cuda/NVIDIA_SLA_cuDNN_Support.txt
cuda/lib64/libcudnn.so
cuda/lib64/libcudnn.so.7
cuda/lib64/libcudnn.so.7.1.2
cuda/lib64/libcudnn_static.a

然后把这些文件按照类别放到cuda中

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

然后就好了。另外ubuntu中还支持通过deb包安装。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值