当前位置:   article > 正文

ubuntu安装cudnn_ubuntu 安装cudnn

ubuntu 安装cudnn

1. 下载cudnn10.1:

官网:https://developer.nvidia.com/rdp/cudnn-archive

可以下载两种安装文件:

我使用的是第一种安装方法.

①按上图下载的是cudnn-*tgz的压缩包

解压下载的文件,可以看到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*

 

执行安装,其实就是拷贝头文件和库文件并给予权限;

 

查看cudnn版本

在终端输入

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

如果出现下图所示版本信息,说明安装成功。

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/花生_TL007/article/detail/627460
推荐阅读
相关标签
  

闽ICP备14008679号