欢迎光临散文网 会员登陆 & 注册

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

2022-05-28 18:30 作者:雪域小白鸠  | 我要投稿

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和驱动对应关系看下图

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检查驱动是否安装完好


安装CUDA(和centos步骤一样)

这里以CUDA11.1为例子,10.x包括后面的安装都是一样的,依葫芦画瓢即可

centos也是一样的,就不单独写专栏了


首先安装CUDA的run文件

sudo ./cuda_11.1.xxx_xxx.xx_linux.run

这里需要注意,因为我们已经安装过nvidia显卡驱动,所以这里展开cuda的安装选项的时候,我们需要去掉显卡驱动前面的勾勾


像这样去掉driver前面的X

安装完成之后,我们需要修改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则代表安装成功,恭喜



Ubuntu 18.04/20.04快速安装Nvdia驱动及其CUDA的评论 (共 条)

分享到微博请遵守国家法律