ubuntu安装cuda12.1和cudnn版本
时间: 2025-01-08 08:46:01 浏览: 16
### 安装 CUDA 12.1 和兼容的 cuDNN 版本于 Ubuntu
对于在Ubuntu上安装CUDA 12.1以及与其相匹配的cuDNN版本,操作流程如下:
#### 下载并准备环境
确保系统已更新至最新状态[^2]。
```bash
sudo apt-get update && sudo apt-get upgrade -y
```
移除旧版NVIDIA驱动程序和CUDA工具包,防止冲突发生[^3]。
```bash
sudo apt-get purge nvidia* cuda*
```
添加官方CUDA仓库密钥与源列表文件到Apt配置中以便后续安装稳定版本软件包[^4]。
```bash
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.0-1_all.deb
sudo dpkg -i cuda-keyring_1.0-1_all.deb
```
注意上述命令中的`ubuntu2004`应替换为当前使用的Ubuntu发行代号名称,如`focal`(对应20.04),`jammy`(对应22.04)等。
#### 执行安装过程
执行实际的CUDA安装指令,这里指定安装的是12.1版本[^5]。
```bash
sudo apt-get update
sudo apt-get -y install cuda-12-1
```
完成以上步骤之后重启计算机使更改生效,并验证安装是否成功通过运行nvcc编译器查看其版本信息来确认[^6]。
```bash
reboot now
# After rebooting, check the installation.
nvcc --version
```
针对cuDNN库部分,则需前往[NVIDIA官方网站](https://developer.nvidia.com/rdp/cudnn-archive)下载适用于所安装CUDA版本对应的cuDNN tarball压缩包。解压后将其中的内容复制到相应目录下即可[^7]。
```bash
tar -xzvf cudnn-linux-x86_64-*.*.*_*-cuda*.tgz
sudo cp cuda/include/* /usr/local/cuda/include/
sudo cp cuda/lib64/* /usr/local/cuda/lib64/
```
最后记得设置好环境变量PATH和LD_LIBRARY_PATH指向新安装好的路径位置,这样可以方便调用相关命令行工具及链接动态库文件[^8]。
```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
```
阅读全文