欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Ubuntu16安装驱动,cuda和cuDNN

程序员文章站 2022-06-06 11:20:19
...

Ubuntu16安装驱动,cuda和cuDNN

系统环境:ubuntu16.04

首先查看显卡版本:lspci | grep -i vga

Ubuntu16安装驱动,cuda和cuDNN
即 “GP106 [GeForce GTX 1060]”

1. 安装显卡驱动

01.准备工作,选择版本

参考博文:https://blog.csdn.net/yinwangde/article/details/89439648
去官网找显卡对应的驱动版本,

英伟达官网 http://www.nvidia.cn/Download/index.aspx?lang=cn

根据电脑配置选择相应的驱动程序,在此本系列选择GeForce GTX 1060 ,操作系统 Linux 64 bit ,如下图所示。
这里写图片描述
Ubuntu16安装驱动,cuda和cuDNN搜索推荐的版本:
Ubuntu16安装驱动,cuda和cuDNN所以此处安装显卡驱动450

02 禁用nouveau

ubuntu 16.04默认安装了第三方开源的驱动程序nouveau,安装nvidia显卡驱动首先需要禁用nouveau,不然会碰到冲突的问题,导致无法安装nvidia显卡驱动。指令如下:

1.1打开编辑配置文件

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

打开文件,在最后添加如下两行:

blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist rivatv
blacklist nvidiafb

1.2 更新系统修改(当时没做这步)

sudo update-initramfs -u      

输入指令后重启系统(一定要重启),确保到位。

1.3 验证nouveau是否已禁用

lsmod | grep nouveau

后没有任何输出说明已禁用。
Ubuntu16安装驱动,cuda和cuDNN

03 下载驱动文件并指令安装

  1. 若默认有安装的驱动,先卸载系统中已存在的驱动sudo apt-get remove --purge nvidia*
  2. 按 Ctrl+Alt+F1 进入命令行界面:
  3. 关闭图形界面,必须关闭: sudo service lightdm stop
  4. 查看当前文件夹下的所有文件,有无下载的驱动包:ls
  5. 给驱动run文件赋予可执行权限:sudo chmod a+x NVIDIA-Linux-x86_64-390.25.run (下载的驱动文件名,按Tab键自动补全)
  6. 安装:sudo ./NVIDIA-Linux-x86_64-xxx.run -no-x-check -no-nouveau-check -no-opengl-files ,其中:

-no-x-check:安装驱动时关闭X服务

-no-nouveau-check:安装驱动时禁用nouveau

-no-opengl-files:只安装驱动文件,不安装OpenGL文件

在安装过程中会出现:

1、he distribution-provided pre-install script failed! Are you sure you want to continue? 选择 yes 继续。

2、Would you like to register the kernel module souces with DKMS? This will allow DKMS to automatically build a new module, if you install a different kernel later? 选择NO继续

3、Would you like to run the nvidia-xconfigutility to automatically update your x configuration so that the NVIDIA x driver will be used when you restart x? Any pre-existing x confile will be backed up. 选择 No 继续
4、gcc版本过高? 选择忽略ignore

  1. 安装成功后,在命令行输入:sudo service lightdm start //重启图形界面, 按Ctrl+Alt+F7返回图形界面

  2. nvidia-smi //检测是否安装成功,出现nvidia信息成功
    nvidia-setting //检测是否安装成功,弹出设置框成功

(二)安装CUDA

https://blog.csdn.net/weixin_42718092/article/details/86016973
nvidia与cuda需相匹配

CUDA提供两种安装方式:package manager安装和runfile安装, package manager 安装方式相对简单一些,但是我在阅读别人博客的过程中发现选择这种方式在安装过程中问题可能多一点,失败的概率较大。为了减少不必要的麻烦我选择runfile安装方式。

00.卸载旧版本的cuda

卸载原来的cuda8.0(注意:不需要卸载显卡驱动,不要给自己找麻烦)

sudo /usr/local/cuda-9.0/bin/uninstall_cuda_9.0.pl

卸载之后,会发现 /usr/local/cuda-8.0目录下任然有文件存在,这是cudnn文件,所以还需要将cuda-8.0文件删除干净:

sudo rm -rf /usr/local/cuda-9.0

01.下载与安装包

下载cuda安装包:cuda官网下载(https://developer.nvidia.com/cuda-toolkit-archive),根据系统信息选择对应的版本,runfile安装的话最后一项要选择 runfile文件,如下图片。

当时网页无法下载,使用下面的两行命令下载的。
Ubuntu16安装驱动,cuda和cuDNN

#下载,时间很长需要两小时
wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run
#获取权限
sudo chmod a+x cuda_11.0.2_450.51.05_linux.run
#安装
sudo sh cuda_11.0.2_450.51.05_linux.run

依次输入accept,进入下图,不选择driver驱动,将两个“X”按Enter键取消选择,然后install。如下
Ubuntu16安装驱动,cuda和cuDNNUbuntu16安装驱动,cuda和cuDNN
Ubuntu16安装驱动,cuda和cuDNN然后ctrl+c,接着按下面的步骤选择:

accept

n(不要安装driver)

一路y

y

y

安装完成后,设置环境变量。
Ubuntu16安装驱动,cuda和cuDNN

02.设置环境变量

打开~/.bashrc文件:

sudo gedit ~./bashrc

将以下内容写入到~/.bashrc尾部:

export PATH=/usr/local/cuda-11.0/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH

保存后,更新文件使该配置生效:一定要执行啊 ,如果同时打开多个终端,执行过后,重新打开,否则后面的错误也是不断的.

source ~/.bashrc

03. 验证是否安装成功

终端输入:nvcc --version 会输出CUDA的版本信息

安装cmake3.17.5

cmake3.17不需要编译,没有编译的文件,下载好后移动到/opt下添加环境变量即可。

  1. 将下载好的安装包解压移动到opt下
sudo mv /home/zfb/cmake-3.17.5-Linux-x86_64 /opt
  1. 安装依赖库:
sduo apt-get install build-essential libssl-dev

可以换源

sudo gedit /etc/apt/sources.list
sudo apt update
  1. 添加环境变量
 sudo gedit .bashrc

将cmake路径添加至末尾:

export PATH=/opt/cmake-3.17.5-Linux-x86_64/bin:$PATH

sudo ~/.bashrc
  1. 查看cmake版本
cmake --version

Ubuntu16安装驱动,cuda和cuDNN

相关标签: Linux安装