linux安装cudnn8.0
时间: 2023-10-19 07:31:19 浏览: 85
要在 Linux 上安装 cuDNN 8.0,您需要按照以下步骤进行操作:
1. 首先,请确保您已经安装了 NVIDIA GPU 驱动程序和 CUDA 工具包。cuDNN 是 CUDA 深度神经网络库,它依赖于 CUDA。
2. 登录到 NVIDIA 开发者网站,下载适用于 Linux 的 cuDNN 8.0 库文件。您可以在 https://developer.nvidia.com/cudnn 上找到 cuDNN 的下载链接。
3. 下载库文件后,将其解压缩到您选择的目录中。例如,假设您将其解压缩到 /usr/local/cudnn 目录中。
```shell
$ tar -xzvf cudnn-8.0-linux-x64-v8.x.x.x.tgz
```
4. 进入解压缩后的目录:
```shell
$ cd cuda
```
5. 将 cuDNN 文件复制到 CUDA 安装目录中:
```shell
$ sudo cp -P include/cudnn.h /usr/local/cuda/include
$ sudo cp -P lib64/libcudnn* /usr/local/cuda/lib64
```
6. 设置库文件的权限:
```shell
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
```
7. 最后,您需要更新您的环境变量 `LD_LIBRARY_PATH`,以使系统能够找到 cuDNN 库。打开您的 `.bashrc` 或 `.bash_profile` 文件,并添加以下行:
```shell
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
```
保存文件并退出。然后,运行以下命令使更改生效:
```shell
$ source ~/.bashrc
```
现在,您已经成功安装了 cuDNN 8.0。您可以在 CUDA 加速的深度学习框架(如 TensorFlow、PyTorch 等)中使用它了。
阅读全文