ubuntu 16.04安装NVIDIA显卡驱动以及cuda、cudnn,以及解决Ubuntu循环登陆问题等

1,180 阅读2分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

一、安装显卡驱动

1. 禁用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

如果什么都没有输出就说明禁用成功

2. 下载驱动文件并安装

查看显卡版本

lspci | grep -i nvidia

查看电脑适合的版本

sudo apt-cache search nvidia*
  1. 下载驱动文件 驱动官网地址:www.nvidia.cn/Download/in… 一般情况下载.run文件安装
  2. 进入命令行界面 按下Ctrl+alt+f1 登陆账户即可
  3. 关闭图形界面
 sudo service lightdm stop
  1. 卸载原有驱动
sudo apt remove nvidia-*  # --purge

如果使用以上命令卸载后运行nvidia-smi任然有显卡信息输出说明没有卸载成功,可使用以下方法卸载(使用runfile文件安装的驱动)

sudo sh ./NVIDIA-Linux-x86_64-xxx.run --uninstall
  1. 安装新驱动
sudo chmod  a+x NVIDIA-Linux-x86_64-xxx.run
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. 安装完成后,开启图形界面
sudo service lightdm start
  1. 验证是否安装成功
nvidia-smi

3. 循环登陆问题解决

如果安装好之后出现循环登陆问题,请从第二步开始,卸载刚安装过的驱动,重新安装一遍(安装命令记得后面加参数),

如果不成,多试几遍,多多踩坑 

二、 安装cuda

1. 下载安装文件

developer.nvidia.com/cuda-toolki… 按照官方命令安装即可

sudo sh cuda_10.0.130_410.48_linux.run

注意:在安装过程中会询问一下东西,在问是否安装驱动时,如果前面已经安装过了,就选择否(最好是按前面的方法手动安装)

2. 设置环境变量,添加路径

sudo gedit ~/.bashrc

在文末添加一下信息

export PATH=/usr/local/cuda-10.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

激活环境

source ~/.bashrc 

3. ubuntu 安装多版本cuda,及切换版本方法

文章传送门

ubuntu 安装多版本cuda,及切换版本方法

三、安装cudnn

1.下载安装文件

cudnn下载 官网地址:developer.nvidia.com/rdp/cudnn-a…

注意:这里需要下载和cuda版本对应的文件

2. 安装cudnn

1. 解压下载文件

tar -xvf cudnn-10.0-linux-x64-xxx.tgz

2. 复制并给予权限

解压后是名为cuda的一个文件夹,执行以下命令

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

3. 查看cudnn版本

【Ubuntu】Linux查看CUDA和cudnn版本方法

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2