Ubuntu24版 最新安装CUDA驱动方式
文章目录
- 背景介绍
- 安装CUDA
- （一）选择合适的CUDA版本
- 1. 进入CUDA下载页面
- 2. 选择适合的安装包
- 3. 获取安装命令
- （二）执行安装命令
- 4. 配置环境变量
- 5. 验证CUDA安装
- 结尾
欢迎来到 全干工程师的运维之路 系列专栏。
无论是前/后端的开发者，若是团队不大，运维方面的技术就成了必须要接触的知识。更多知识请关注本专栏哦
背景介绍
上一步我们完成了显卡驱动的安装，那么接下来就是必不可少的CUDA，如果驱动还没安装好，请参考:显卡驱动
NVIDIA CUDA是一个由NVIDIA推出的并行计算平台和编程模型，可以利用NVIDIA显卡进行高效的计算任务加速。如果你正在使用Ubuntu操作系统并且希望在你的GPU上进行深度学习、科学计算等任务，那么正确安装CUDA驱动和工具包是必不可少的。
这篇博客将详细介绍如何在Ubuntu上安装CUDA驱动，从选择合适的CUDA版本到安装过程中的每一步，帮你轻松完成整个安装流程。
安装CUDA
（一）选择合适的CUDA版本
首先，我们要选择与显卡驱动兼容的CUDA版本。我显卡的驱动版本是 550.120，因此我可以选择安装 CUDA 12 版本。一般来说，当前大部分显卡驱动版本都在 525 以上，可以安装 CUDA 12 版本。如果你的显卡较旧，可以根据显卡驱动版本查找适配的CUDA版本。
1. 进入CUDA下载页面
前往NVIDIA官方CUDA工具包下载页面，你会看到多个版本的CUDA工具包。这里建议选择 Latest（最新版本）。
2. 选择适合的安装包
在选择下载CUDA工具包的版本后，会弹出一个平台选择窗口。根据你的操作系统平台选择相应的选项。以下是几个常见的选择：
- 操作系统：选择 Ubuntu。
- 版本：选择你当前使用的Ubuntu版本。
- 安装类型：可以选择 deb (local) 或 deb (network)，具体取决于你的需求。
3. 获取安装命令
选择好平台和安装类型后，页面下方会出现一段安装命令。我们只需按步骤逐步执行这些命令。
（二）执行安装命令
按照上面的步骤完成安装，我们先完成前三步：
第四步可能出现出现GPG key未安装的提示，他的报错是这样的，可以执行给出的倒数第三句指令来安装GPG key：
4. 配置环境变量
CUDA安装完成后，我们需要设置一些环境变量。在~/.bashrc文件中加入以下内容：
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}}
然后执行以下命令使改动生效：
source ~/.bashrc
5. 验证CUDA安装
最后，检查CUDA是否成功安装。你可以通过运行以下命令来验证：
nvcc --version
如果一切顺利，你应该会看到CUDA的版本信息。
结尾
经过以上步骤，你应该已经顺利地在Ubuntu系统上安装好了CUDA工具包。此时，你可以开始利用GPU进行高效的并行计算了。不论是深度学习训练、图像处理还是其他计算密集型任务，CUDA都能为你提供强大的加速能力。
一起变得更强😊