ubuntu 安装 cuda

ubuntu 安装 cuda

本篇文章将介绍ubuntu 安装 CUDA Toolkit

CUDA Toolkit 是由 NVIDIA（英伟达）公司开发的一个软件工具包，用于支持并优化 GPU（图形处理器）上的并行计算和高性能计算。它提供了一组库、编译器、开发工具和 API（应用程序编程接口），使开发者能够在 NVIDIA GPU 上编写和执行并行计算任务，从而加速各种应用程序的运行速度。

主要功能和组件包括：

1. CUDA 编程模型：CUDA Toolkit 提供了一种并行计算模型，允许开发者使用类似于 C 语言的语法来编写 GPU 上的代码。开发者可以使用 CUDA C/C++ 编写核心计算代码，称为 GPU 内核函数（Kernel Functions），并在 GPU 上并行执行这些函数。
2. CUDA Runtime API：CUDA 提供了一组运行时库和 API，开发者可以使用这些函数来管理 GPU 内存、启动 GPU 内核函数、同步和通信等。
3. CUDA 编译器：CUDA 编译器将 CUDA C/C++ 代码转换为针对特定 NVIDIA GPU 架构的汇编代码。这些汇编代码会与 CPU 代码一起组成一个可执行文件。
4. GPU 加速库：CUDA Toolkit 提供了一些高性能计算库，如 cuBLAS（基础线性代数子程序库）、cuFFT（快速傅里叶变换库）、cuDNN（深度神经网络库）等，这些库能够在 GPU 上加速相应的计算任务。
5. 并行计算支持：CUDA Toolkit 充分利用 GPU 的并行计算能力，可以用于各种任务，如科学计算、机器学习、深度学习、图像处理等。

初

希望能写一些简单的教程和案例分享给需要的人

环境与设备

系统：ubuntu
设备：Nvidia GeForce RTX 4090

在官网找安装方式

访问

https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64

选择后就和下图一样：我这里是 Linux ubuntu 22.04 所以我按这个来选。

复制下面的 Base Installer

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
sudomv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.2.1/local_installers/cuda-repo-ubuntu2204-12-2-local_12.2.1-535.86.10-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2204-12-2-local_12.2.1-535.86.10-1_amd64.deb
sudocp /var/cuda-repo-ubuntu2204-12-2-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudoapt-get update
sudoapt-get-yinstall cuda

一个个复制过去执行

使用

wget

命令下载一个名为

cuda-ubuntu2204.pin

的文件。这个文件是用来设置 APT 软件源优先级的。

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin

将刚刚下载的

cuda-ubuntu2204.pin

文件移动到

/etc/apt/preferences.d/

目录下，并将文件名更改为

cuda-repository-pin-600

。这个操作会设置 CUDA 软件源的优先级为 600，以确保系统正确选择合适的软件源。

sudomv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600

使用

wget

命令下载 CUDA Toolkit 的本地安装包（

.deb

文件），以便在本地系统中进行安装。

wget https://developer.download.nvidia.com/compute/cuda/12.2.1/local_installers/cuda-repo-ubuntu2204-12-2-local_12.2.1-535.86.10-1_amd64.deb

使用

dpkg

命令安装刚刚下载的 CUDA 本地安装包。

sudo dpkg -i cuda-repo-ubuntu2204-12-2-local_12.2.1-535.86.10-1_amd64.deb

将 CUDA 软件源中的密钥文件复制到

/usr/share/keyrings/

目录下，这是用于验证软件源的 GPG 密钥。

sudocp /var/cuda-repo-ubuntu2204-12-2-local/cuda-*-keyring.gpg /usr/share/keyrings/

更新系统的软件包信息，以便获取最新的可用软件包列表。

sudoapt-get update

使用

apt-get

命令安装 CUDA Toolkit。

-y

参数表示自动确认安装过程中的提示。

sudoapt-get-yinstall cuda

安装完成后，执行命令

dir /usr/local/

执行命令后可以看到 cuda ，说明安装了，接下来配置一下文件

修改配置文件：vim ~/.bashrc

vim ~/.bashrc

在末尾加入下面代码：我这里是12.2，所以我加的是这个路径 12.2，大家自行更换

exportLD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-12.2/lib64
exportPATH=$PATH:/usr/local/cuda-12.2/bin
exportCUDA_HOME=$CUDA_HOME:/usr/local/cuda-12.2

加入配置文件后，执行代码更新

source ~/.bashrc

更新完成后，我们执行 nvcc -V 来查看是否安装完成

nvcc -V