ubuntu22.04安装 NVIDIA CUDA Toolkit 和 cuDNN 库
时间: 2024-10-11 21:04:04 浏览: 38
ubuntu20.04显卡驱动,CUDA 安装和卸载,cudnn安装.pdf
5星 · 资源好评率100%
在Ubuntu 22.04上安装NVIDIA CUDA Toolkit和cuDNN库,你需要遵循一些步骤:
1. **更新系统**:
首先确保你的系统是最新的,运行以下命令:
```
sudo apt update && sudo apt upgrade
```
2. **添加CUDA PPA (个人软件包存档)**:
对于非官方支持的版本,如CUDA 11.x在22.04上,你需要添加PPA:
```
sudo add-apt-repository ppa:graphics-drivers/ppa
```
接着,更新软件列表:
```
sudo apt-get update
```
3. **安装CUDA Toolkit**:
安装指定版本的CUDA,例如CUDA 11.6:
```
sudo apt install nvidia-cuda-toolkit=11.6.*
```
4. **安装cuDNN**:
对于cuDNN,通常CUDA会包含对应版本的cuDNN。你可以通过`nvidia-smi`检查CUDA是否安装并包含了cuDNN。如果需要手动安装,访问NVIDIA官网下载cuDNN,并从其压缩文件中解压到合适的目录,然后将路径添加到环境变量`LD_LIBRARY_PATH`中。
5. **设置环境变量**:
更新系统的环境变量配置,确保它们可以在终端中访问:
```bash
echo "export PATH=/usr/local/cuda/bin:$PATH" >> ~/.bashrc
echo "export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH" >> ~/.bashrc
source ~/.bashrc
```
6. **验证安装**:
使用`nvcc --version`检查CUDA编译器版本,`cudnnVersion()`或在Python中导入`cupy`库来验证cuDNN。
阅读全文