重新安装Ubuntu双系统后,需要用GPU跑一些东西,所以要先装一下显卡驱动
本人环境:RTX 2070 + Ubuntu 18.04(经测试,本文同样适用于Ubuntu16.04)
目录
0. 第一步
先更新一下系统:
sudo apt-get update
sudo apt-get upgrade
1. 安装显卡驱动
如果你用的是 Ubuntu 桌面版(即带图形化界面),那么去 Nvidia 官网下载驱动就是你遇到的第一个陷阱。当你试图安装时,系统会报错说你该关闭 X window 或者降低系统的 run level。大意就是在安装显卡驱动前,你得先关停系统的图形化界面。
1.1. 如果你用的是 Ubuntu 桌面版
那么用 apt-get 安装驱动最简单,但你需要点耐心,因为驱动下载的速度比较慢。
添加开源显驱仓库
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
安装一些依赖
sudo apt-get install dkms synaptic build-essential
打开系统的 Software & Update 工具,选择 Additional Drivers,此时你会看到可以下载的显卡驱动。
驱动安装完成后,根据提示重启电脑即可。
1.2. 如果你用的是 Ubuntu 服务器版
- 从官网下载显卡的最新驱动版本。
cd
到驱动安装包所在的目录并用sh NVIDIA-Linux-x86_64-version.run
运行安装包,注意把version替换成你下载的驱动版本号。- 用
sudo reboot
重启电脑后,驱动就安装完成了。
1.3. 检查驱动是否安装成功
在命令行输入下述命令,看到显卡当前的状态例如温度、显存占用情况等,说明驱动安装成功了。
nvidia-smi
顺便说个小技巧,如果你在跑模型时想监控显卡的温度,也可以用这个命令:
watch -n 1 nvidia-smi # 1表示每1秒刷新一次