在Ubuntu上配置CUDA基础环境

一、安装驱动

1、检查硬件支持的驱动列表：

# 桌面版本
sudo ubuntu-drivers list
# 服务器版本
sudo ubuntu-drivers list --gpgpu

可以看到这样的驱动名称列表：

nvidia-driver-470
nvidia-driver-470-server
nvidia-driver-535
nvidia-driver-535-open
nvidia-driver-535-server
nvidia-driver-535-server-open
nvidia-driver-550
nvidia-driver-550-open
nvidia-driver-550-server
nvidia-driver-550-server-open

2、安装驱动

自动安装推荐的驱动：

sudo ubuntu-drivers install

或者选择一个驱动版本安装，例如 nvidia-driver-535-server，截取后缀 535-server：