ubuntu18下安装nvidia驱动以及cuda

一、安装二、卸载

cuda包中包含了驱动安装程序，所以可以通过官网一起安装
cuda及nvidia驱动官网下载

选择自己想要的cuda版本，根据自己的系统选择对应的安装包

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda-repo-ubuntu1804-11-4-local_11.4.0-470.42.01-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1804-11-4-local_11.4.0-470.42.01-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu1804-11-4-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

而后按照官方的指导一步步安装。

Fri Feb 18 06:49:38 2022       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 470.42.01    Driver Version: 470.42.01    CUDA Version: 11.4     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  NVIDIA GeForce ...  Off  | 00000000:AF:00.0 Off |                  N/A |
| 16%   29C    P8    14W / 250W |      1MiB / 11019MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|  No running processes found                                                 |
+-----------------------------------------------------------------------------+

通过nvidia-smi就可以打印出结果。

卸载显卡驱动

sudo apt-get remove nvidia-*

清除无用包

sudo apt-get purge nvidia*
sudo apt-get purge libnvidia*
sudo apt-get autoremove

清空缓存

sudo apt-get clean

直到通过以下的查看命令看不到任何相关包

sudo dpkg --list | grep nvidia-*

这时候就卸载干净了，最后重启开始安装。