Linux下怎么安装cuDNN

这篇“Linux下怎么安装cuDNN”文章的知识点大部分人都不太理解，所以小编给大家总结了以下内容，内容详细，步骤清晰，具有一定的借鉴价值，希望大家阅读完这篇文章能有所收获，下面我们一起来看看这篇“Linux下怎么安装cuDNN”文章吧。

NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。NVIDIA cuDNN可以集成到更高级别的机器学习框架中。

下载

之前下载和安装的驱动和cuda版本如下

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 418.43       Driver Version: 418.43       CUDA Version: 10.1     |
|-------------------------------+----------------------+----------------------+

所以我们只需要安装对应的cudnn即可

重命名格式

cp cudnn-10.1-linux-x64-v7.5.0.56.solitairetheme8 cudnn-10.1-linux-x64-v7.5.0.56.tgz

解压

tar -xzvf cudnn-10.1-linux-x64-v7.5.0.56.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 /usr/local/cuda/lib64/libcudnn*

以上就是关于“Linux下怎么安装cuDNN”这篇文章的内容，相信大家都有了一定的了解，希望小编分享的内容对大家有帮助，若想了解更多相关的知识内容，请关注亿速云行业资讯频道。