关于NVIIDA驱动

如果你的设备有一张NVIIDA的显卡，并且你想要将它利用起来，就需要安装驱动

安装

安装显卡驱动的方法有很多种，这里推荐两种比较简单的方法进行安装

从Ubuntu的软件更新中安装

打开 软件和更新 -> 附加驱动程序

选择你需要的版本，点击应用更改即可

autoinstall

打开终端输入

ubuntu-drivers devices

会自动检测你的系统硬件，并给你推荐可以安装的驱动

如果同意安装推荐的版本

sudo ubuntu-drivers autoinstall

就可以自动安装推荐版本的驱动，安装完成之后重启电脑 reboot

重启之后驱动就会正常工作

其他安装方法

也可以从NVIDIA官网下载对应版本驱动手动安装

这种方法网上教程很多，并且较为繁琐，不是很推荐

CUDA

1
2
3
4
5
6
7
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/10.2/Prod/local_installers/cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda
  • 将 CUDA 添加至环境变量
    • sudo gedit ~/.bashrc 打开 ~/.bashrc 文件，在最后加入
1
2
export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64{LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

CUDNN

1
sudo dpkg -i libcudnn8_8.0.5.39-1+cuda10.2_amd64.deb

测试

1
nvidia-smi

查看驱动等的信息

或者

1
2
source ~/.bashrc
nvcc --version

正确输出CUDA版本信息

  • 驱动相关配置完成

参考

知乎