Ubuntu 18.04安装显卡驱动及配置CUDA开发环境

Ubuntu 18.04 CUDA

1.禁用 nouveau

1.1 查看公版驱动状态-无输出为已禁用

lsmod | grep nouveau

1.2 禁用公版驱动

 sudo gedit /etc/modprobe.d/blacklist.conf

在打开的文件最后加入以下两行并保存：

blacklist nouveau
options nouveau modeset=0

1.3 禁执行更新内核

sudo update-initramfs -u 

重启查看公版驱动状态，应该为无输出。

2.安装Nidia驱动

列出所有可用驱动

ubuntu-drivers devices

选择适用版本安装

sudo apt install nvidia-driver-470

3.安装CUDA

自己去官网下载适合版本的CUDA, 为了Pytorch选的CUDA 10.2
终端运行安装：

sudo sh cuda_10.2.89_440.33.01_linux.run

编辑环境变量

sudo gedit ~/.bashrc

在后面加入并保存文件

export CUDA_HOME=/usr/local/c