1 Ubuntu驱动和Cuda安装

1.1 Nvidia显卡驱动安装

方式一: 通过官方源安装

sudo apt update
# 安装ubuntu驱动自动管理库
sudo apt install ubuntu-drivers-common
# 查看当前主机的显卡和推荐的驱动版本
sudo ubuntu-drivers devices
# 安装驱动,注意需要安装-open版本,否则可能会报错, -open是开源版本;
# 较新的西安看尽量选择开源版本。如果是10系列老显卡可以选择闭源版本;
sudo apt install nvidia-driver-610-open

方式二: 通过nvidia源安装

在nvidia官方下载对应版本的驱动,注意,需要根据环境选择正确的版本 cuda官方下载地址 使用deb(network)最方便; 官方的截图

# 下载安装nvidia官方的证书和源使用的deb安装包
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
# 更新源
sudo apt update
# 安装驱动
sudo apt-get install -y nvidia-open

1.2 安装cuda

见上面的方式二,安装完nvidia的源之后,通过:

sudo apt update
sudo apt list |grep nvidia-open
sudo apt install -y open nvidia-open

多版本共存和版本切换

截图2 将需要版本的usr/local/cuda-*/bin目录添加到PATH;

警告

不同的cuda版本对驱动的版本有最低版本要求; 所以如果升级cuda版本之前需要升级驱动版本;