Ubuntu16.04+RTX2070s 安装显卡驱动+CUDA+CUDNN+Pytorch

安装Ubuntu之后由于没有显卡驱动,默认使用了Nouveau,显示器分辨率极低

1、安装显卡驱动

一般可以使用ppa源安装或者自己从官网下,我在使用ppa添加源的时候一直没反应,可能是网络原因,所以这里选择了第二种:

从NVIDIA官网:https://www.geforce.cn/drivers选择合适的版本下载,放在英文目录下(有可能在文本模式下没法显示中文)

如果之前有安装驱动,需要先卸载旧版本驱动,没有就不用:

$sudo apt-get remove nvidia-*

$sudo apt-get autoremove  (加上才能卸载干净原驱动)

输入CTRL+ALT+F1进入文本模式

关闭显示服务:

$sudo service lightdm stop

权限:

$chmod a+x NVIDIA-Linux-x86_64-418.43.run

这里注意用自己的安装包名称

$sudo ./NVIDIA-Linux-x86_64-418.43.run -no-x-check -no-nouveau-check -no-opengl-files

以下三个最好加上,否则还是可能循环登录

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

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

    -no-opengl-files 安装时只装驱动文件,不安装Opengl

关于禁用Nouveau,还有一种方法是在/etc/modprobe.d/blacklist.conf的最后一行加上

blacklist nouveau

然后使其生效:

$sudo update-initramfs -u

使用这个命令看是否有输出,如果没有证明禁用成功

$lsmod | grep nouveau

我两种都做了,目前没有出现什么问题

安装过程中:

在NVIDIA驱动安装过程中,依次的选项为:

选:accept(注:白色为选中,不是红色)

The distribution-provided pre-install script failed … …

选:Continue installation

Would you like to run the nvidia-xconfig utility to automatically update your X Configuration file so set the NVIDIA X driver will be used when you restart X?

选:NO

Install 32-Bit compatibility libraries?

选:NO

启动显示服务

$sudo service lightdm restart(将自动跳转到桌面)

查看Nvidia驱动是否安装成功

$nvidia-smi

显示信息的话表示安装成功

2、安装CUDA和CUDNN

CUDA:

去英伟达官网下载cuda安装包:https://developer.nvidia.com/cuda-toolkit-archive

跟据你的平台选择好后会出现安装命令,依照其执行即可,比如,

选择合适的版本,我选择了.run形式的安装包,下载好后执行

$sudo sh cuda_10.1.243_418.87.00_linux.run

在安装过程中,会让你选择是否安装驱动,这里有于我们已经装过了,所以选择不装驱动

然后设置环境:在~/.bashrc文件最后添加下面几行

LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64

PATH=$PATH:/usr/local/cuda-10.1/bin

CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.1

然后输入

$nvcc --version

如果出现如下版本信息,证明安装成功:

nvcc: NVIDIA (R) Cuda compiler driver

Copyright (c) 2005-2019 NVIDIA Corporation

Built on Sun_Jul_28_19:07:16_PDT_2019

Cuda compilation tools, release 10.1, V10.1.243

也可以这样进行测试:

$cd /usr/local/cuda-10.1/samples/1_Utilities/deviceQuery

$sudo make

$sudo ./deviceQuery

如果最后输出PASS,则安装成功

CUDNN:

依然是官网下载安装包:https://developer.nvidia.com/rdp/cudnn-archive

这里选择了对应CUDA版本的对应linux版,文件格式是.solitairetheme8,需要先转成tgz格式再解压,直接重命名即可。

解压下载的文件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*

查看cudnn版本信息:

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

出现版本信息则安装成功

#define CUDNN_MAJOR 7

#define CUDNN_MINOR 6

#define CUDNN_PATCHLEVEL 5

--

#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#include "driver_types.h"

你可能感兴趣的:(Ubuntu16.04+RTX2070s 安装显卡驱动+CUDA+CUDNN+Pytorch)