重邮虚拟机环境配置(三)如何在虚拟机上安装驱动

(1)检查 GPU 是否被识别

lspci | grep -i nvidia

(2)安装 NVIDIA 驱动,使用 Ubuntu-drivers 自动安装

sudo apt update sudo ubuntu-drivers autoinstall

(3)重启虚拟机

sudo reboot

(4)检查驱动是否加载

nvidia-smi

(5)检查 CUDA 支持

sudo apt install nvidia-cuda-toolkit nvcc --version