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

在官网找安装方式

访问

bash 复制代码
https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64

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

复制下面的 Base Installer

bash 复制代码
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 软件源优先级的。

bash 复制代码
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,以确保系统正确选择合适的软件源。

bash 复制代码
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600

使用 wget 命令下载 CUDA Toolkit 的本地安装包(.deb 文件),以便在本地系统中进行安装。

bash 复制代码
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 本地安装包。

bash 复制代码
sudo dpkg -i cuda-repo-ubuntu2204-12-2-local_12.2.1-535.86.10-1_amd64.deb

将 CUDA 软件源中的密钥文件复制到 /usr/share/keyrings/ 目录下,这是用于验证软件源的 GPG 密钥。

bash 复制代码
sudo cp /var/cuda-repo-ubuntu2204-12-2-local/cuda-*-keyring.gpg /usr/share/keyrings/

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

bash 复制代码
sudo apt-get update

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

bash 复制代码
sudo apt-get -y install cuda

安装完成后,执行命令

bash 复制代码
dir /usr/local/

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

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

bash 复制代码
vim ~/.bashrc

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

bash 复制代码
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

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

bash 复制代码
source ~/.bashrc

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

bash 复制代码
nvcc -V
相关推荐
暮暮七36 分钟前
hive和spark读写分区表NULL列
大数据·linux·hive·hadoop·sql·spark
EasyDSS1 小时前
互联网视频推拉流EasyDSS视频直播点播平台视频转码有哪些技术特点和应用?
运维·服务器·音视频·rtmp推流·视频转码
HengYuan_Tech1 小时前
gitlab ssh-key 绑定
运维·ssh·gitlab
有来技术1 小时前
Linux 服务器安装 Docker - CentOS 9 (Stream)
linux·服务器·docker
jekc8682 小时前
Centos 相关网络配置
linux·网络·centos
DADIAN_GONG2 小时前
How to install tree on Centos? what is difference between apt and yum?
linux·运维·centos
Karoku0662 小时前
【docker集群应用】Docker常用命令
运维·数据库·docker·容器
夏沫の浅雨4 小时前
imx-6ULL uboot 移植
linux·uboot·nxp·6ull
C++忠实粉丝4 小时前
计算机网络之应用层协议HTTP
linux·网络·c++·网络协议·tcp/ip·计算机网络·http