感谢各位看友来访,更多最新技术专栏文章尽在作者『 全栈工程师修炼指南』星球,加入一起学习交流吧!
2.CUDA Toolkit 安装
什么是 CUDA?
CUDA(
Compute Unified Device Architecture)是英伟达(NVIDIA)推出的并行计算平台和编程模型。它允许开发者利用 NVIDIA GPU 的并行计算能力进行通用计算任务,而不仅限于图形处理。CUDA 提供了一个统一的编程接口和软件环境,使开发者可以使用类似于 C 的编程语言来编写并行计算程序。
官方地址: https://developer.nvidia.com/cuda-toolkit
主要特点和功能:
并行计算能力:CUDA 允许开发者利用 GPU 的大规模并行处理单元来加速各种类型的计算任务,包括科学计算、机器学习、深度学习、图像处理等。
编程模型:CUDA 提供了一个基于 C/C++ 的编程模型,开发者可以使用类似于传统 CPU 编程的方式来编写 GPU 加速的代码。此外,CUDA 还支持 Fortran 和 Python 等语言的接口。
软件工具集:NVIDIA 提供了包括 CUDA Toolkit 在内的一系列开发工具和库,帮助开发者在 CUDA 平台上进行开发、调试和优化。
性能优化:CUDA 具备优化性能的工具和技术,包括并行算法、内存管理、线程调度等,使得开发者能够充分利用 GPU 的硬件并行性能。
生态系统:CUDA 已经成为广泛使用的 GPU 计算平台,支持多种操作系统和编程环境,为科学计算和工业应用提供了强大的支持。
总体而言,CUDA 已经成为科学计算和各种数据密集型应用的重要工具之一,通过利用 GPU 的并行计算能力,显著提高了计算速度和效率。
[!note] 温馨提示
每个 CUDA 工具包版本都有与之对应开发 NVIDIA GPU 驱动程序版本,如作者安装的显卡驱动版本为610.43.02,则对应的 CUDA Toolkit 版本为CUDA 13.3 Update 1.
CUDA工具包和相应的驱动程序对应参考链接:CUDA Toolkit and Corresponding Driver Versions
安装步骤
步骤 01.根据显卡驱动版本选择 CUDA Toolkit 版本。
步骤 02.访问 Nvidia CUDA DOWNLOAD 地址根据操作系统进行选择,例如 操作系统 Linux > 架构 X86_64 > 发行版 Ubuntu > 版本 26.04 > 安装类型 deb(local) 或者 runfile(local)
步骤 03.根据 CUDA Toolkit Installer Type 提示进行下载安装,例如以 deb(local) 方式安装,执行的脚本
cd /opt/soft/nvidia
# 下载 Nvidia 软件源
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2604/x86_64/cuda-ubuntu2604.pin
sudo mv cuda-ubuntu2604.pin /etc/apt/preferences.d/cuda-repository-pin-600
# 下载 deb 本地安装包
wget https://developer.download.nvidia.com/compute/cuda/13.3.1/local_installers/cuda-repo-ubuntu2604-13-3-local_13.3.1-610.43.02-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2604-13-3-local_13.3.1-610.43.02-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2604-13-3-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda-toolkit-13-3
步骤 04.若如任何错误的情况下,便可设置CUDA-Toolkit环境变量,在~/.bashrc或者/etc/profile目录中添加环境变量,这样每次在登录服务器便会自动注入到执行终端,非常重要否则 nvcc 命令不能执行,例如:
# 创建 CUDA toolkit 环境变量到 /etc/profile.d/ 目录下的 nvidia-cuda.sh 文件中
tee /etc/profile.d/nvidia-cuda.sh <<'EOF'
# CUDA toolkit ENV
export CUDA_HOME=/usr/local/cuda export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:${CUDA_HOME}/lib64 export PATH=${CUDA_HOME}/bin:${PATH}
EOF
# 赋予执行权限
chmod +x /etc/profile.d/nvidia-cuda
# 应用环境变量
source /etc/profile.d/nvidia-cuda
步骤 05.最后执行如下命令,验证 CUDA Toolkit 使用安装成功,若成功命令执行后将显示如下:
$ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2026 NVIDIA Corporation
Built on Tue_Jun_09_02:43:40_PM_PDT_2026
Cuda compilation tools, release 13.3, V13.3.73
Build cuda_13.3.r13.3/compiler.38244171_0
至此,CUDA Toolkit 在 Ubuntu 26.04 LTS 服务器中安装成功。
温馨提示:若文章代码块中存在乱码或不能复制,请联系作者,也可通过文末的阅读原文链接,加入知识星球中阅读,原文链接:https://articles.zsxq.com/id_8no2nl5cm9dh.html
学习:作者知识星球
『 全栈工程师修炼指南』星球,主要涉及全栈工程师(Full Stack Development)实践文章,持续更新包括但不限于企业SecDevOps和网络安全等保合规、安全渗透测试、编程开发、云原生(Cloud Native)、物联网工业控制(IOT)、人工智能Ai,从业书籍笔记,人生职场认识等方面资料或文章。
935