显卡驱动—cuda+cudnn

1. 卸载原有驱动（没装跳过）

    方法一、通用sudo apt-get remove –purge nvidia*
    方法二、 非官网驱动sudo apt-get remove --purge nvidia（版本全称，如我的nvidia-384.130）
    方法三、官网驱动 sh ./nvidia.run --uninstall

2. 禁用nouveau

安装NVIDIA需要先禁用系统自带的驱动

    打开文件sudo gedit /etc/modprobe.d/blacklist.conf

    在文本末尾添加

    blacklist nouveau
    option nouveau modeset=0

    无视warn

    wq， 执行sudo update-initramfs -u

    检测，lsmod | grep nouveau，没有任何输出说明禁用成功
3. 安装NVIDIA显卡驱动

    查看支持的驱动版本

    sudo apt-cache search nvidia*，一般安装大于等于最大的版本就好，我安装的是418.87

    下载NVIDIA驱动，

    官网地址

    禁用X服务（最好有一台备用电脑，因为这一步会黑屏...因为我是采用ssh连接，所以对我没影响）

    sudo service lightdm stop
    ctr+alt+F1

    给文件赋执行权限并cd到下载目录

    cd ~
    sudo chmod a+x NVIDIA-Linux-x86_64-384.130.run

    安装

    sudo sh ./NVIDIA-Linux-x86_64-375.20.run –no-opengl-files

    重启x服务

    sudo service lightdm start
    ctr+alt+F7

    检测是否成功 nvidia-smi

 

 

cuda+cudnn

1.cudnn的安装
    sudo ./cuda_10.1.243_418.87.00_linux.run
    sudo gedit ~/.bashrc

    添加到最后：
    export PATH=/usr/local/cuda-10.1/bin:$PATH
    export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

    source ~/.bashrc
    nvcc -V

2.cudnn的安装

（1）安装cudnn

解压下载的文件，可以看到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*

（2）查看cudnn版本

在终端输入

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