文章目录
欢迎来到 全干工程师的运维之路 系列专栏。
无论是前/后端的开发者,若是团队不大,运维方面的技术就成了必须要接触的知识。更多知识请关注本专栏哦
背景介绍
上一步我们完成了显卡驱动的安装,那么接下来就是必不可少的CUDA,如果驱动还没安装好,请参考:显卡驱动
NVIDIA CUDA是一个由NVIDIA推出的并行计算平台和编程模型,可以利用NVIDIA显卡进行高效的计算任务加速。如果你正在使用Ubuntu操作系统并且希望在你的GPU上进行深度学习、科学计算等任务,那么正确安装CUDA驱动和工具包是必不可少的。
这篇博客将详细介绍如何在Ubuntu上安装CUDA驱动,从选择合适的CUDA版本到安装过程中的每一步,帮你轻松完成整个安装流程。
安装CUDA
(一)选择合适的CUDA版本
首先,我们要选择与显卡驱动兼容的CUDA版本。我显卡的驱动版本是 550.120,因此我可以选择安装 CUDA 12 版本。一般来说,当前大部分显卡驱动版本都在 525 以上,可以安装 CUDA 12 版本。如果你的显卡较旧,可以根据显卡驱动版本查找适配的CUDA版本。