Ubuntu18.04安装nvidia驱动（1060显卡）、Cuda10.1

一、先安装Nvidia驱动程序

先确定对应的显卡驱动，然后下载

https://www.geforce.cn/drivers

(驱动下载命令

wget -c http://cn.download.nvidia.com/XFree86/Linux-x86_64/410.78/NVIDIA-Linux-x86_64-410.78.run

)

按照以下教程进行

https://blog.csdn.net/yhaolpz/article/details/71375762

或：自动安装命令：第一次使用的时候失败，可能是因为当时没有配好所需gcc和cmake，后来手动安装的时候装上了

ubuntu-drivers autoinstall

 

禁用 nouveau

配置环境变量

按照第五步关闭桌面服务进入文本模式

（Ubuntu18.04以上命令部分不可用，改为：

停止可视化桌面：

$ sudo telinit 3       

 

之后会进入一个新的命令行会话，使用当前的用户名密码登录

在相应路径下安装NVIDIA驱动(安装文件也可为.sh后缀，如果提示没有权限使用sudo)：

$ bash NVIDIA-Linux-x86_64-384.111.bin

干净的系统安装时需要gcc和cmke:(gcc在后面安装cuda时有版本要求，查看https://docs.nvidia.com/cuda/)

sudo apt install gcc
sudo apt install cmake

在安装结束后，在命令行输入一下命令重启，NVIDIA驱动即可安装成功：

$ sudo reboot

）

安装驱动

安装驱动如果提示GCC版本不对，按照以下教程更换版本

https://blog.csdn.net/u012968002/article/details/71389713

测试，nvidia-smi，出现显卡信息则为成功

 

二、cuda安装

下载地址，https://developer.nvidia.com/cuda-toolkit-archive

因为安装nVidia驱动的时候安装了最新的版本，而且卸载失败，所以抱着试试看的心态尝试安装cuda10.1，结果安装成功。

在安装时需要gcc和g++，之前没有装g++，安装一下

$ sudo apt install g++      

cuda10.1的安装界面与看到的教程不太一样，更加方便，由于之前安装了nVidia的驱动，在安装cuda时不选择安装nVidia驱动以免出错。

三、cudnn安装

在官网下载安装包，需要注册登录才能下载。选择适合自己的版本

下载完成后解压并进入文件夹：

    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

 

完成