Ubuntu虚拟机中如何安装显卡驱动

175
2025/3/18 15:32:52
栏目: 云计算
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

在Ubuntu虚拟机中安装显卡驱动的步骤如下:

1. 检查系统是否已安装GPU驱动

首先,打开终端并输入以下命令来检查系统是否已安装GPU驱动:

nvidia-smi

如果命令报错,说明驱动未安装。

2. 禁用Nouveau驱动

Nouveau是一个开源的NVIDIA显卡驱动,通常在系统启动时自动加载。为了安装官方的NVIDIA驱动,需要先禁用Nouveau驱动。

  • 使用文本编辑器(如vim)打开/etc/modprobe.d/blacklist.conf文件:

    sudo vim /etc/modprobe.d/blacklist.conf
    
  • 在文件末尾添加以下内容:

    blacklist nouveau
    options nouveau modeset=0
    
  • 保存并退出编辑器。

  • 更新系统以应用更改:

    sudo update-initramfs -u
    
  • 重启系统:

    sudo reboot
    
  • 重启后,使用以下命令检查Nouveau驱动是否已被禁用:

    lsmod | grep nouveau
    

    如果没有输出,说明禁用成功。

3. 安装NVIDIA驱动

方法一:使用ubuntu-drivers工具

  • 查看系统推荐的驱动版本:

    ubuntu-drivers devices
    
  • 安装推荐的驱动版本:

    sudo ubuntu-drivers autoinstall
    
  • 安装完成后,重启系统:

    sudo reboot
    
  • 使用nvidia-smi命令验证驱动是否安装成功:

    nvidia-smi
    

方法二:手动下载并安装NVIDIA驱动

  1. 访问NVIDIA官网,根据你的显卡型号下载对应的驱动程序(通常是.run文件)。

  2. 关闭所有图形界面,进入命令行界面(例如,使用Ctrl+Alt+F1)。

  3. 禁用图形界面服务(例如,gdm):

    sudo systemctl stop gdm
    
  4. 切换到下载的驱动文件所在目录,并赋予执行权限:

    cd /path/to/your/driver/file
    sudo chmod +x NVIDIA-Linux-x86_64-xxx.run
    
  5. 运行驱动安装程序:

    sudo ./NVIDIA-Linux-x86_64-xxx.run --no-opengl-files --no-x-check --no-nouveau-check
    

    注意:--no-opengl-files参数禁用OpenGL文件安装,--no-x-check--no-nouveau-check参数在安装过程中跳过一些检查。

  6. 安装完成后,启动图形界面服务:

    sudo systemctl start gdm
    
  7. 使用nvidia-smi命令验证驱动是否安装成功:

    nvidia-smi
    

4. 安装cuDNN(可选)

如果你需要使用CUDA,还需要安装cuDNN库。

  1. 下载对应版本的cuDNN库(通常与CUDA版本匹配)。

  2. 解压下载的cuDNN压缩包:

    sudo dpkg -i cudnn-local-repo-ubuntu2004-8.6.0.163_1.0-1_amd64.deb
    
  3. 更新软件库:

    sudo cp /var/cudnn-local-repo-ubuntu2004-8.6.0.163/cudnn-local-B0FE0A41-keyring.gpg /usr/share/keyrings/
    sudo apt-get update
    
  4. 安装cuDNN库:

    sudo apt-get install libcudnn8=8.9.0.131-1+cuda11.8 libcudnn8-dev=8.9.0.131-1+cuda11.8 libcudnn8-samples=8.9.0.131-1+cuda11.8
    

完成以上步骤后,你的Ubuntu虚拟机应该已经成功安装了显卡驱动,并且可以使用nvidia-smi命令查看显卡信息。如果遇到问题,可以参考相关日志文件或搜索错误信息以获取更多解决方案。

辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读: PyTorch在Ubuntu上的安装步骤是什么