Ubuntu安装cuda

瞻邈2024-08-17 18:26

本文详细介绍了在 Ubuntu 系统上安装 CUDA 的全过程。从安装前的系统要求和准备工作，到具体的安装步骤，包括下载 CUDA 安装文件、处理依赖关系、执行安装命令以及配置环境变量等。旨在为需要在 Ubuntu 中安装 CUDA 以进行深度学习、图形计算等工作的用户提供清晰、准确且可操作的指导。

1. 查看驱动版本

1.1. 检查环境

输入

bash 复制代码

uname -a

和

bash 复制代码

cat /etc/lsb-release

进行操作系统检查

1.2. 检测显卡信息

输入

bash 复制代码

lspci | grep VGA

和

bash 复制代码

sudo lshw -C video

查看显卡信息，我们可以看到Nvidia Tesla T4显卡

2. 查看支持驱动版本的cuda版本

1. CUDA 12.6 Release Notes --- Release Notes 12.6 documentation

3. 安装cuda

3.1. apt安装

安装依赖项

bash 复制代码

sudo apt update -y && sudo apt upgrade -y && sudo apt install vim gcc g++ make python3-pip -y

安装完成后建议最好进行一次系统重启

bash 复制代码

sudo apt install ubuntu-drivers-common

根据自己的需要，选择相应的cuda版本，例如：

bash 复制代码

sudo apt install cuda-11-8

3.2. 下载安装

下载地址 https://developer.nvidia.com/cuda-toolkit-archive

bash 复制代码

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.0-1_all.deb
sudo dpkg -i cuda-keyring_1.0-1_all.deb
sudo apt-get update
sudo apt-get -y install cuda

4. cuda版本切换

bash 复制代码

update-alternatives --display cuda

bash 复制代码

update-alternatives --config cuda

参考文献

Ubuntu Server 22.04 LTS安装CUDA[机器学习环境搭建]机器学习环境搭建第三篇：基于Ubuntu和 - 掘金

opencv - How to change CUDA version - Stack Overflow