怎么安装本地的cudnn安装包
时间: 2023-10-28 10:01:40 浏览: 52
安装 CUDNN 之前,需要先确保已经安装了 NVIDIA 的显卡驱动和 CUDA 工具包。
安装步骤如下:
1. 首先,下载对应版本的 CUDNN 安装包,可以从 NVIDIA 的官网上下载:https://developer.nvidia.com/cudnn
2. 下载后,解压缩安装包,可以看到包含以下文件:
```
cudnn.h
libcudnn.so -> libcudnn.so.x.x.x
libcudnn.so.x.x.x
libcudnn_static.a
```
3. 将解压后的文件复制到 CUDA 的安装目录下,一般位于 `/usr/local/cuda/`。
```
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*
```
4. 最后,更新环境变量:
```
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
```
或者将其添加到 `.bashrc` 文件中,使其永久生效:
```
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc
```
完成上述步骤后,即可在代码中使用 CUDNN 加速深度学习算法。