本篇文章将介绍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
加入配置文件后,执行代码更新
source ~/.bashrc
更新完成后,我们执行 nvcc -V 来查看是否安装完成
nvcc -V