【机器学习】Ubuntu系统下显卡驱动卸载及重装

目录

背景

驱动卸载

驱动安装

CUDA驱动安装

安装nvidia-smi


背景

这里包含显卡驱动和CUDA驱动,在如下场景下,我们需要卸载显卡驱动并重新安装。

  1. 在某些情况下需要对显卡驱动进行升级
  2. 某些情况下(如重启,或者调整系统配置等),系统出问题导致显卡驱动掉了,需要重新安装

在上述过程中,经常会出现一些问题导致显卡驱动安装失败,本文参考官方给定的一些方式安装。

驱动卸载

  1. 使用apt卸载相应驱动(不用care是否存在)

    复制代码
    sudo apt --purge remove nvidia*
    sudo apt --purge remove "*cublas*" "cuda*"
    sudo apt --purge remove "*nvidia*
    sudo apt autoremove
  2. 上述操作完成后,并不一定生效,还要使用安装后提供的工具卸载存在的驱动 这里的文件不一定存在,取决于一开始的cuda 安装方式,如果不存在,直接跳过即可

    复制代码
    # 这里会弹出界面让你确认卸载内容,根据提示操作即可,选择上所有组件进行卸载
    sudo /usr/local/cuda-**.*/bin/cuda-uninstaller
    
    # 完成上述操作后,可能会有残留文件需要阐述
    sudo rm -rf /usr/local/cuda*

卸载完驱动后,需要重启。

驱动安装

CUDA驱动安装

官网参考

以CUDA 12.3 为例。注意,如果使用官方run文件的形式安装可能会报错,推荐如下方式安装。

复制代码
wget <https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin>
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget <https://developer.download.nvidia.com/compute/cuda/12.3.2/local_installers/cuda-repo-ubuntu2204-12-3-local_12.3.2-545.23.08-1_amd64.deb>
sudo dpkg -i cuda-repo-ubuntu2204-12-3-local_12.3.2-545.23.08-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2204-12-3-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda-toolkit-12-3

完成上述安装后,nvcc -V命令应该可以正常工作,但nvidia-smi无法正常工作。

安装nvidia-smi

复制代码
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update

sudo apt search nvidia-driver
# 选择上述内容列表中自己需要的/最新的
sudo apt install -y nvidia-driver-550

安装完成后,重启后nvidia-smi即可生效。

相关推荐
szxinmai主板定制专家3 小时前
基于ZYNQ MPSOC图像采集与压缩系统总体设计方案
linux·arm开发·人工智能·嵌入式硬件·fpga开发
liulilittle3 小时前
TCP UCP:基于卡尔曼滤波的BBR增强型拥塞控制算法
linux·网络·c++·tcp/ip·算法·c·通讯
AI医影跨模态组学3 小时前
J Thorac Oncol(IF=20.8)广东省人民医院钟文昭教授团队:基于影像组学的支持向量机区分驱动肺腺癌进展的分子事件
人工智能·深度学习·机器学习·论文·医学·医学影像·影像组学
tianrun12344 小时前
Ubuntu 24.04 安装 Fcitx5 + Rime + 搜狗词库(替代 IBus 与 Linux 搜狗输入法)
linux·运维·ubuntu
wxytxdy4 小时前
通过猜数字游戏学习Shell脚本的分支、循环编写
linux·学习
AI医影跨模态组学4 小时前
Radiol Artif Intell 中山大学肿瘤防治中心放疗科:基于连续MRI的深度学习模型预测局部晚期鼻咽癌患者生存期
人工智能·深度学习·论文·医学·医学影像·影像组学
YIN_尹4 小时前
【Linux 系统编程】手撕一个简易版的shell命令行解释器
android·linux·运维
小猫咪014 小时前
Linux 软链接和硬链接详解:ln 命令背后的 inode 原理
linux
Daydream.V4 小时前
深度学习常见激活函数详解(Sigmoid/Tanh/ReLU/Leaky ReLU/Swish/GELU)优缺点+场景对比
人工智能·深度学习
小脑斧1235 小时前
从入门到精通:Linux 进程间通信 IPC 全解析|管道、共享内存、信号量、消息队列实战
linux·运维·服务器