这篇“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”这篇文章的内容,相信大家都有了一定的了解,希望小编分享的内容对大家有帮助,若想了解更多相关的知识内容,请关注天达云行业资讯频道。