cuDNN环境变量设置ubuntu
时间: 2024-07-21 09:01:14 浏览: 132
cuDNN(CUDA Deep Neural Network Library)是一个由NVIDIA提供的用于加速深度学习运算的库。在Ubuntu上安装和配置cuDNN通常涉及到几个步骤:
1. **安装依赖**:首先确保你的系统已经安装了CUDA,因为cuDNN是CUDA的一部分。可以访问NVIDIA官网下载对应版本的CUDA。
2. **下载cuDNN**:从NVIDIA Developer网站下载cuDNN,选择适合你CUDA版本的版本。例如,如果你有CUDA 11.x,就下载cuDNN 8.x版本。
3. **解压并移动文件**:将下载的.tgz文件解压到一个合适的路径,比如`/usr/local/cuda`下的cudnn文件夹。
4. **设置环境变量**:
- 打开终端,编辑`.bashrc`或`.bash_profile`文件(通常是`~/.bashrc`):
```
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CUDA_HOME/lib64
export CUDNN_INCLUDE_DIR=$CUDA_HOME/include
export CUDNN_LIBRARY=$CUDA_HOME/lib64/libcudnn.so
```
5. **加载新环境**:运行`source ~/.bashrc`(或者重启终端)使更改生效。
6. **验证安装**:你可以通过命令行运行`nvidia-smi`和`cdnnVerifyVersion`来检查cuDNN是否安装成功,并查看其版本信息。
注意:如果遇到权限问题,可能需要添加sudo前缀。此外,上述步骤可能会因Ubuntu的具体版本和个人喜好而略有差异,建议查阅最新文档或参考NVIDIA官方教程。
阅读全文