Ubuntu 18.04/20.04快速安装Nvdia驱动及其CUDA

1 首先检查基础环境是否ok
sudo apt install build-essential
(里面包含了大部分基础环境包括g++,make等,如果没有网络请参考我的离线配置文章)
下载nvidia驱动,最好放到/home/user/Download下方便找
https://www.nvidia.cn/Download/index.aspx?lang=cn(选择linux64位的下载)
下载cuda,也放到Download下,cuda需要看对应版本
https://developer.nvidia.com/cuda-toolkit-archive
cuda和驱动对应关系看下图

cudnn的下载页面和对应关系
https://developer.nvidia.com/rdp/cudnn-archive
2 禁用nouveau并重启(安装Nvidia显卡的官方驱动和系统自带的nouveau驱动冲突)
sudo gedit /etc/modprobe.d/blacklist.conf
在打开文本的最后一行添加:
blacklist nouveau
options nouveau modeset=0
更新,重启:
sudo update-initramfs -u
reboot
重启后再次查看是否禁用nouveau
lsmod | grep nouveau
如果没有任何输出代表禁用ok
3 安装显卡驱动(和centos步骤一样)
首先
sudo init3 ---
进入单用户模式
此时屏幕会进入terminal黑屏背景
按照提示login用户之后
cd进入你的驱动所在目录,以我的为例子我放在了/home/user/Download下
cd /home/user/Download
添加执行权限
sudo chmod a+x NVIDIA-Linux-x86_64-xxx.run
执行安装
sudo ./NVIDIA-Linux-x86_64-xxx.run -no-opengl-files
这里要注意如果安装提示x服务错误,类似“You appear to be running an X server; please exit X before installing.”,请在后面继续加参数变成如下的命令再运行
sudo ./NVIDIA-Linux-x86_64-xxx.run -no-opengl-files -no-x-check
安装过程中
32bit兼容包选择, 这里要注意选择NO,不然后面就会出错。
X-configurtion的选择页面NO
安装完成后
sudo init5 切换回桌面模式
终端输入nvidia-smi检查驱动是否安装完好
4 安装CUDA(和centos步骤一样)
这里以CUDA11.1为例子,10.x包括后面的安装都是一样的,依葫芦画瓢即可
centos也是一样的,就不单独写专栏了
首先安装CUDA的run文件
sudo ./cuda_11.1.xxx_xxx.xx_linux.run
这里需要注意,因为我们已经安装过nvidia显卡驱动,所以这里展开cuda的安装选项的时候,我们需要去掉显卡驱动前面的勾勾

安装完成之后,我们需要修改bashrc
vim ~/.bashrc
之后在最下方添加下列两行
export PATH=$PATH:/usr/local/cuda-11.1/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.1/lib64
然后source一下
source ~/.bashrc
你在哪个用户下安装的驱动和cuda,对应的用户下bashrc就需要添加,如果不确定请在root和你当前账户下都添加一下
5 编译SAMPLE(测试CUDA是否安装成功)
cd /usr/local/cuda-11.1/samples/
编译
make all - j100(内核数,可以往大的写)
编译完成后
cd bin/x86_64/linux/release/
然后测试cuda是否test ok
./deviceQuery
输出结尾看到Result=PASS则代表安装成功,恭喜
