cudnn下载及安装linux
时间: 2024-09-24 14:01:07 浏览: 147
CUDA Deep Neural Network (cuDNN)是一个由NVIDIA专为深度学习加速而设计的GPU计算库。要在Linux上下载并安装cuDNN,你可以按照以下步骤操作:
1. **检查CUDA版本**:
确保已经安装了CUDA,因为cuDNN需要与其匹配。打开命令行,输入`nvcc --version`查看CUDA版本。
2. **访问cuDNN官网**:
访问CUDA Toolkit的cuDNN页面:https://developer.nvidia.com/cudnn
3. **选择合适的cuDNN版本**:
根据你的CUDA版本选择相应的cuDNN版本。例如,如果你的CUDA是11.x,那么你需要cuDNN 8.x。
4. **下载cuDNN文件**:
下载cuDNN的.tgz或.zst压缩包。注意,cuDNN一般只提供预编译好的二进制包,不需要源码编译。
5. **解压和移动到正确的目录**:
解压下载的文件,然后将`lib64`(对于64位系统)或`lib`(32位系统)下的内容复制到你的CUDA `include` 和 `lib` 目录下。通常路径类似 `/usr/local/cuda/include` 和 `/usr/local/cuda/lib64`。
6. **设置环境变量**:
编辑CUDA的配置文件(`~/.bashrc`, `~/.bash_profile`, 或者`/etc/profile`),添加如下内容:
```sh
export CUDA_PATH=/usr/local/cuda
export CUDNN_LIBRARY=$CUDA_PATH/lib64/libcudnn.so.8
export PATH=$CUDA_PATH/bin:$PATH
```
7. **加载cuDNN**:
刷新shell或运行 `source ~/.bashrc`(或其他配置文件),然后运行 `nvidia-smi`验证是否成功加载了cuDNN。
8. **测试安装**:
使用Python等深度学习框架如PyTorch或TensorFlow的cuDNN支持功能进行测试,看是否能够正常工作。
阅读全文