CUDNN(CUDA深度神经网络库)是NVIDIA提供的一个用于深度学习的加速库,它可以提供深度神经网络的加速运算和训练。以下是在Ubuntu系统上安装CUDNN的指南:
下载CUDNN: 首先,你需要从NVIDIA官网上下载适用于你的CUDA版本的CUDNN。确保你下载的版本和你的CUDA版本兼容。
解压CUDNN文件: 下载完成后,将CUDNN压缩文件解压到一个目录中。
复制文件到CUDA目录: 进入解压后的CUDNN目录,将其中的文件复制到你的CUDA安装目录中。通常情况下,CUDA安装在/usr/local/cuda目录下。
sudo cp -r include/* /usr/local/cuda/include/
sudo cp -r lib64/* /usr/local/cuda/lib64/
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda/bin:$PATH
保存修改后,执行以下命令使修改生效:
source ~/.bashrc
nvcc -V
如果成功安装,会显示CUDA和CUDNN的版本信息。
以上就是在Ubuntu系统上安装CUDNN的简单指南,希望对你有所帮助。