ubuntu

ubuntu中cudnn安装的方法是什么

小亿
165
2024-02-21 09:58:20
栏目: 编程语言

在Ubuntu中安装cuDNN(CUDA Deep Neural Network library)可以按照以下步骤进行:

  1. 首先,需要去NVIDIA官网注册并下载cuDNN的压缩包,地址为:https://developer.nvidia.com/cudnn

  2. 下载完成后,将压缩包解压缩到指定目录,例如解压到/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是根据你下载的具体版本号和系统架构来确定的,需要根据实际情况进行更改。

  1. 配置环境变量,在.bashrc.zshrc等文件中添加如下路径:
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
  1. 最后,运行以下命令使变量生效:
source ~/.bashrc

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

0
看了该问题的人还看了