cudnn linux 安装
时间: 2023-08-27 19:07:24 浏览: 90
要在 Linux 上安装 cuDNN(CUDA Deep Neural Network library),请按照以下步骤进行操作:
1. 首先,确保您已经安装了 CUDA Toolkit。如果还没有安装 CUDA Toolkit,请先安装它。cuDNN 是 CUDA Toolkit 的一部分。
2. 访问 NVIDIA 官方网站(https://developer.nvidia.com/cudnn)并注册或登录您的账号。
3. 在 cuDNN 下载页面中,选择与您的 CUDA 版本和操作系统相匹配的 cuDNN 版本进行下载。您需要下载三个文件:cuDNN Runtime Library、cuDNN Developer Library 和 cuDNN Code Samples。
4. 下载完成后,使用以下命令将 cuDNN 解压到适当的目录中:
```
tar -xzvf cudnn-<version>.tgz
```
5. 进入解压后的 cuDNN 目录:
```
cd cuda
```
6. 将 cuDNN 文件复制到 CUDA Toolkit 的安装目录中。默认情况下,CUDA Toolkit 安装在 `/usr/local/cuda` 目录下。执行以下命令进行复制:
```
sudo cp include/cudnn*.h /usr/local/cuda/include
sudo cp lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
```
7. 最后,添加 cuDNN 到环境变量中。打开 `~/.bashrc` 文件(或者 `~/.zshrc` 文件,如果您在使用 Zsh),并添加以下内容:
```
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
```
保存文件并使配置生效:
```
source ~/.bashrc
```
现在,cuDNN 库应该已经成功安装在您的 Linux 系统上,您可以在深度学习项目中使用它了。