linux安装cuda和cudnn deb
时间: 2025-01-07 18:23:12 浏览: 4
### 使用 deb 包在 Linux 上安装 CUDA 和 cuDNN
#### 安装前准备
为了确保顺利安装,建议先确认系统环境满足最低需求并更新软件源。可以执行以下命令来刷新本地包索引:
```bash
sudo apt-get update && sudo apt-get upgrade -y
```
#### 下载适用于操作系统的 CUDA Toolkit DEB (Network) 文件
访问 NVIDIA 官方网站获取适合当前 Linux 发行版的 CUDA 版本链接。对于基于 Debian 的发行版本如 Ubuntu, 可以直接下载 .deb 文件。
例如,针对特定版本的 CUDA toolkit:
```bash
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
```
接着配置 APT 源以便于后续安装过程更便捷[^1]:
```bash
wget https://developer.download.nvidia.com/compute/cuda/12.0/local_installers/cuda-repo-ubuntu2004-12-0-local_12.0.0-520.30.06-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-12-0-local_12.0.0-520.30.06-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-12-0-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
```
完成上述设置之后就可以正式开始安装 CUDA 工具链了:
```bash
sudo apt-get install -y cuda
```
#### 配置环境变量
为了让系统能够识别新安装好的 CUDA 编译器和其他工具,默认情况下可能还需要手动调整 PATH 和 LD_LIBRARY_PATH 环境变量。编辑 `~/.bashrc` 或者 `/etc/profile`, 添加如下几行内容到文件末尾[^2] :
```bash
export PATH=/usr/local/cuda-12.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.0/lib64\
${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
```
使更改生效:
```bash
source ~/.bashrc
```
#### 安装 cuDNN 库
由于官方并没有提供独立的 cuDNN `.deb` 包形式分发渠道,因此仍然推荐采用 tar 形式的压缩包来进行部署。不过在此之前应该已经完成了 CUDA 的基础环境搭建工作。按照之前提到的方式下载对应版本号的 cuDNN 后解压至指定目录即可.
注意:如果希望获得更好的兼容性和性能表现,则应当选择与已安装 CUDA 版本相匹配的 cuDNN 版本。
阅读全文