ubuntu中cudnn安装的方法是什么-开发者知识库平台

ubuntu中cudnn安装的方法是什么

ubuntu

700

2024/3/6 9:58:20

栏目: 编程语言

在Ubuntu中安装cuDNN（CUDA Deep Neural Network library）可以按照以下步骤进行：

首先，需要去NVIDIA官网注册并下载cuDNN的压缩包，地址为：https://developer.nvidia.com/cudnn
下载完成后，将压缩包解压缩到指定目录，例如解压到/usr/local目录下：

tar -xzvf cudnn-10.0-linux-x64-v7.6.5.32.tgz
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*

注意：这里的cudnn-10.0-linux-x64-v7.6.5.32.tgz是根据你下载的具体版本号和系统架构来确定的，需要根据实际情况进行更改。

配置环境变量，在.bashrc或.zshrc等文件中添加如下路径：

export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

最后，运行以下命令使变量生效：

source ~/.bashrc

完成以上步骤后，cuDNN就成功安装到系统中了。可以检查是否安装成功，以及cuDNN的版本等信息。

辰迅云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>

ubuntu中cudnn安装的方法是什么

最新知识库

相关标签