pip 安装 cudnn linux
时间: 2023-11-18 12:53:48 浏览: 156
首先,要安装 cuDNN,需要先下载对应版本的 cuDNN 库文件,然后将其解压到相应的目录下。具体步骤如下:
1. 下载 cuDNN 库文件
首先需要到 NVIDIA 官网上下载对应版本的 cuDNN 库文件。下载地址为:https://developer.nvidia.com/rdp/cudnn-download 。在下载页面中,需要选择相应的 cuDNN 版本、操作系统和 CUDA 版本,然后点击“Download”按钮进行下载。
2. 解压 cuDNN 库文件
下载完成后,将下载的文件解压到相应的目录下。假设解压后的文件夹名为“cudnn”,则可以使用以下命令将其解压到 /usr/local/cuda 目录下:
```
$ tar -xzvf cudnn-10.1-linux-x64-v7.6.5.32.tgz
$ sudo cp -P cuda/include/cudnn*.h /usr/local/cuda/include
$ sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
```
其中,cudnn-10.1-linux-x64-v7.6.5.32.tgz 是下载的 cuDNN 库文件名,需要根据实际情况进行修改。
3. 配置环境变量
最后,需要将 cuDNN 库文件所在的目录添加到 LD_LIBRARY_PATH 环境变量中,以便系统能够找到相应的库文件。可以在 ~/.bashrc 文件中添加以下内容:
```
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
```
然后执行以下命令使配置生效:
```
$ source ~/.bashrc
```
阅读全文