liunx安装cudnn
时间: 2023-08-31 11:10:01 浏览: 37
要在Linux上安装cuDNN,您可以按照以下步骤进行操作:
1. 下载cuDNN:
在NVIDIA的开发者网站上,您需要先注册一个帐号。然后,根据您的CUDA版本和操作系统,在cuDNN下载页面上选择适当的版本。
2. 解压文件:
使用`tar`命令解压下载的文件。进入下载目录并执行以下命令:
```
tar -xzvf cudnn-<version>-linux-x64-v8.x.x.tgz
```
将`<version>`替换为您下载的cuDNN版本号。
3. 将文件复制到CUDA安装目录:
进入解压后的文件夹,并将其中的文件复制到CUDA的安装目录。通常,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. 配置环境变量:
打开您的shell配置文件(例如`~/.bashrc`或`~/.zshrc`),并将以下行添加到文件末尾:
```
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
```
保存文件并执行以下命令使更改生效:
```
source ~/.bashrc
```
5. 验证安装:
运行以下命令来验证cuDNN是否正确安装:
```
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
```
如果成功安装cuDNN,您将看到类似以下输出:
```
#define CUDNN_MAJOR 8
#define CUDNN_MINOR 2
#define CUDNN_PATCHLEVEL 0
```
这些步骤应该能够帮助您在Linux上成功安装cuDNN。请确保您已经正确安装了CUDA,并使用与cuDNN版本兼容的CUDA版本。