cudnn 安装 linux
时间: 2023-10-28 07:02:18 浏览: 94
要在Linux上安装cuDNN(CUDA Deep Neural Network Library),请按照以下步骤进行操作:
1. 前提条件:
- 先确保你已经安装了NVIDIA的显卡驱动和CUDA工具包。cuDNN是建立在CUDA的基础上的,所以需要先安装CUDA。
- 确认你的显卡型号是否被cuDNN支持。你可以在NVIDIA的官方网站上找到cuDNN的兼容性列表。
2. 下载cuDNN软件包:
- 访问NVIDIA Developer网站,注册一个账户(如果还没有)并登录。
- 转到cuDNN下载页面:https://developer.nvidia.com/rdp/cudnn-download
- 选择适合你系统的cuDNN版本进行下载。确保选择与你的CUDA版本相匹配的cuDNN版本。
3. 安装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*
```
4. 配置环境变量:
- 编辑你的`.bashrc`文件(或者其他shell的配置文件),添加以下行:
```
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
```
- 保存文件后,重新加载配置或重新启动终端。
5. 验证安装:
- 打开终端,运行以下命令来验证cuDNN是否正确安装:
```
nvcc --version
nvidia-smi
```
- 如果两个命令都能正确执行并显示有关CUDA和显卡的信息,则表示cuDNN安装成功。
请注意,cuDNN的安装可能会因系统和CUDA版本的不同而有所差异。以上步骤提供了一般的安装指南,但具体操作仍需根据你的系统环境进行调整。
阅读全文