Chinaunix首页 | 论坛 | 博客
  • 博客访问: 200669
  • 博文数量: 67
  • 博客积分: 375
  • 博客等级: 一等列兵
  • 技术积分: 525
  • 用 户 组: 普通用户
  • 注册时间: 2011-04-29 22:28
文章分类

全部博文(67)

文章存档

2021年(2)

2019年(1)

2018年(19)

2017年(23)

2016年(4)

2011年(18)

我的朋友

分类: 其他平台

2018-09-02 18:17:05

cudnn是nvidia提供的深度学习接口

一、获取cuDNN

官网下载cudnn5.1: https://developer.nvidia.com/rdp/cudnn-download

(需要注册NVIDIA账号)




cudnn-9.2-linux-x64-v7.2.1.38.tgz

二、安装cuDNN


把头文件和库文件拷贝到cuda的安装目录中,首先确保cuda的安装目录是/usr/local/cuda/,之前安装cuda时,已经默认安装目录为/usr/local/cuda/了

    sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz
    sudo cp cuda/include/cudnn.h /usr/local/cuda/include
    sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64


查看cudnn版本

sudo cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A2

阅读(1384) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~