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
相关推荐
云飞云共享云桌面14 分钟前
8位机械工程师如何共享一台图形工作站算力?
linux·服务器·网络
Peter_chq1 小时前
【操作系统】基于环形队列的生产消费模型
linux·c语言·开发语言·c++·后端
一坨阿亮2 小时前
Linux 使用中的问题
linux·运维
dsywws3 小时前
Linux学习笔记之vim入门
linux·笔记·学习
幺零九零零4 小时前
【C++】socket套接字编程
linux·服务器·网络·c++
wclass-zhengge4 小时前
Docker篇(Docker Compose)
运维·docker·容器
李启柱4 小时前
项目开发流程规范文档
运维·软件构建·个人开发·设计规范
小林熬夜学编程5 小时前
【Linux系统编程】第四十一弹---线程深度解析:从地址空间到多线程实践
linux·c语言·开发语言·c++·算法
力姆泰克5 小时前
看电动缸是如何提高农机的自动化水平
大数据·运维·服务器·数据库·人工智能·自动化·1024程序员节
BPM_宏天低代码5 小时前
低代码 BPA:简化业务流程自动化的新趋势
运维·低代码·自动化