温馨提示×

cudnn ubuntu安装指南

小樊
83
2024-07-05 00:29:28
栏目: 智能运维

CUDNN(CUDA深度神经网络库)是NVIDIA提供的一个用于深度学习的加速库,它可以提供深度神经网络的加速运算和训练。以下是在Ubuntu系统上安装CUDNN的指南:

  1. 下载CUDNN: 首先,你需要从NVIDIA官网上下载适用于你的CUDA版本的CUDNN。确保你下载的版本和你的CUDA版本兼容。

  2. 解压CUDNN文件: 下载完成后,将CUDNN压缩文件解压到一个目录中。

  3. 复制文件到CUDA目录: 进入解压后的CUDNN目录,将其中的文件复制到你的CUDA安装目录中。通常情况下,CUDA安装在/usr/local/cuda目录下。

sudo cp -r include/* /usr/local/cuda/include/
sudo cp -r lib64/* /usr/local/cuda/lib64/
  1. 添加库文件路径到环境变量: 编辑 ~/.bashrc 文件,添加以下几行:
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda/bin:$PATH

保存修改后,执行以下命令使修改生效:

source ~/.bashrc
  1. 验证安装: 打开终端,输入以下命令验证CUDNN是否安装成功:
nvcc -V

如果成功安装,会显示CUDA和CUDNN的版本信息。

以上就是在Ubuntu系统上安装CUDNN的简单指南,希望对你有所帮助。

0