ubuntu 安装 cuda

news2024/10/3 4:44:19

ubuntu 安装 cuda

初
环境与设备
- 在官网找安装方式

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

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

主要功能和组件包括：

CUDA 编程模型：CUDA Toolkit 提供了一种并行计算模型，允许开发者使用类似于 C 语言的语法来编写 GPU 上的代码。开发者可以使用 CUDA C/C++ 编写核心计算代码，称为 GPU 内核函数（Kernel Functions），并在 GPU 上并行执行这些函数。
CUDA Runtime API：CUDA 提供了一组运行时库和 API，开发者可以使用这些函数来管理 GPU 内存、启动 GPU 内核函数、同步和通信等。
CUDA 编译器：CUDA 编译器将 CUDA C/C++ 代码转换为针对特定 NVIDIA GPU 架构的汇编代码。这些汇编代码会与 CPU 代码一起组成一个可执行文件。
GPU 加速库：CUDA Toolkit 提供了一些高性能计算库，如 cuBLAS（基础线性代数子程序库）、cuFFT（快速傅里叶变换库）、cuDNN（深度神经网络库）等，这些库能够在 GPU 上加速相应的计算任务。
并行计算支持：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
sudo mv 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
sudo cp /var/cuda-repo-ubuntu2204-12-2-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install 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，以确保系统正确选择合适的软件源。

sudo mv 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 密钥。

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

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

sudo apt-get update

使用 apt-get 命令安装 CUDA Toolkit。-y 参数表示自动确认安装过程中的提示。

sudo apt-get -y install cuda

安装完成后，执行命令

dir /usr/local/

执行命令后可以看到 cuda ，说明安装了，接下来配置一下文件
在这里插入图片描述

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

vim ~/.bashrc

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

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-12.2/lib64
export PATH=$PATH:/usr/local/cuda-12.2/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-12.2

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