ubuntu24安装cudnn
时间: 2024-11-17 21:11:44 浏览: 31
在Ubuntu 24上安装CUDA和cuDNN通常需要几个步骤,因为这两个库都是深度学习框架如TensorFlow、PyTorch等的重要依赖。以下是安装的基本流程:
1. **更新系统**:
```bash
sudo apt update && sudo apt upgrade
```
2. **安装CUDA**
- 首先去NVIDIA官网下载适用于Ubuntu 24的CUDA版本:https://developer.nvidia.com/cuda-downloads
- 下载并安装CUDA,这会自动处理一些依赖包。
3. **添加CUDA repository(如果有CUDA toolkit for JetPack)**:
```bash
curl https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/keys/NVIDIA-Retail.pub | sudo apt-key add -
echo "deb [arch=amd64] https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/ /" | sudo tee /etc/apt/sources.list.d/cuda.list
```
4. **安装cuDNN**
- 访问NVIDIA cuDNN页面:https://developer.nvidia.com/cudnn
- 根据你的GPU和CUDA版本选择合适的cuDNN版本。通常,你需要下载对应的CUDA Toolkit版本的cuDNN。
- 解压下载的文件,并将其移动到合适的路径,比如`/usr/local/cuda`, 然后运行安装脚本:
```bash
sudo cp cuda_<version>/cuda_<version>.<arch>-linux.<os>-x64.cudnn_<version>.tar.gz /usr/local/
sudo chmod a+x /usr/local/cuda_<version>/cuda_<version>.<arch>-linux.<os>-x64/cuda_installers/cuda_<version>_libcudnn_<version>.so.7.sh
sudo ./cuda_<version>_libcudnn_<version>.so.7.sh
```
5. **验证安装**:
```bash
dpkg -L libcudnn* | grep '/lib/x86_64-linux-gnu'
```
阅读全文