Linux下安装cuDNN具体步骤

Linux下如何安装cuDNN？NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。NVIDIA cuDNN可以集成到更高级别的机器学习框架中，下面为大家分享一下Linux下安装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系统相关内容。想要了解更多Linux相关知识记得关注公众号“良许Linux”，或扫描下方二维码进行关注，更多干货等着你！