ubuntu 下 nvidia 显卡使用默认的开源驱动存在问题,因此使用闭源驱动进行替换。

    使用机器跑深度学习模型时,为了使用GPU的算力,也需要安装 nvidia 的官方驱动。如下:

    1. sudo add-apt-repository ppa:graphics-drivers
    2. sudo apt update -y
    3. ubuntu-drivers devices # 查看推荐安装的驱动
    4. sudo ubuntu-drivers autoinstall # 自动安装推荐的驱动
    5. nvidia-smi # 检查是否安装完成