首页cudnn 安装 linux

cudnn 安装 linux

时间: 2023-10-28 15:02:18 浏览: 109

要在Linux上安装cuDNN（CUDA Deep Neural Network Library），请按照以下步骤进行操作：

前提条件：
- 先确保你已经安装了NVIDIA的显卡驱动和CUDA工具包。cuDNN是建立在CUDA的基础上的，所以需要先安装CUDA。
- 确认你的显卡型号是否被cuDNN支持。你可以在NVIDIA的官方网站上找到cuDNN的兼容性列表。
下载cuDNN软件包：
- 访问NVIDIA Developer网站，注册一个账户（如果还没有）并登录。
- 转到cuDNN下载页面：https://developer.nvidia.com/rdp/cudnn-download
- 选择适合你系统的cuDNN版本进行下载。确保选择与你的CUDA版本相匹配的cuDNN版本。

安装cuDNN：

解压下载的cuDNN压缩包。可以使用以下命令：
```
tar -xzvf cudnn-x.x-linux-x64-v8.x.x.x.tgz
```

将解压后得到的文件复制到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*

配置环境变量：
- 编辑你的.bashrc文件（或者其他shell的配置文件），添加以下行：
```
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
```
- 保存文件后，重新加载配置或重新启动终端。
验证安装：
- 打开终端，运行以下命令来验证cuDNN是否正确安装：
```
nvcc --version
nvidia-smi
```
- 如果两个命令都能正确执行并显示有关CUDA和显卡的信息，则表示cuDNN安装成功。