linux安装cuDnn
时间: 2024-03-25 08:34:55 浏览: 62
cudnn-6.5-linux-x64-v2.tgz
cuDNN(CUDA Deep Neural Network library)是NVIDIA提供的用于深度神经网络加速的库。在Linux上安装cuDNN需要以下步骤:
1. 下载cuDNN:首先,你需要到NVIDIA官网的开发者页面(https://developer.nvidia.com/cudnn)下载适用于你的CUDA版本的cuDNN压缩包。请确保你已经注册为NVIDIA开发者并登录。
2. 解压cuDNN压缩包:使用以下命令解压下载的cuDNN压缩包:
```
tar -xzvf cudnn-<version>.tgz
```
3. 复制文件到CUDA目录:进入解压后的cuDNN目录,使用以下命令将文件复制到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`文件(或者`~/.bash_profile`),添加以下行:
```
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
```
5. 更新环境变量:运行以下命令使环境变量生效:
```
source ~/.bashrc
```
完成以上步骤后,cuDNN就成功安装在你的Linux系统上了。
阅读全文