安装显卡驱动以及CUDA、cudnn

ubantu安装显卡驱动以及CUDA、cudnn

安装显卡驱动

参考链接

1、查看显卡型号

lshw -c video

结果如下：我的型号是GF119M [GeForce 610M]

2、查找显卡驱动

去官网查看显卡驱动
下面是我的电脑对应的驱动版本：

搜索的结果页面看到，我的驱动版本应该是390.132
为了再次确认一遍，你还可以使用这个命令查看你可以使用的驱动：

ubuntu-drivers devices

结果显示

如果结果显示和搜索到的驱动版本一样，推荐也是390.132，那我们可以安心安装390.132版本。但是结果显示的并没有390.132，为了避免出错，建议选择结果显示中的版本号384。

3、安装显卡驱动

1、禁用nouveau

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

在末尾加上下面两句话

blacklist nouveau
options nouveau modeset=0

2、执行命令：

sudo update-initramfs -u  //应用更改

3、重启系统，然后输入验证是否禁用

lsmod | grep nouveau

显示如图则说明成功（图为别人的图，自己忘记截取了）：

4、安装显卡驱动
打开系统设置->软件和更新->附加驱动->选择NVIDIA驱动->应用更改（如图）

4、查看NVIDIA版本检验是否安装完成

nvidia-smi

安装CUDA

安装cuda注意与cudnn版本的匹配
本文安装的是cuda9

1、下载安装文件

下载链接

2、安装cuda

进入下载安装文件的文件夹，然后运行以下命令：

sudo sh cuda_9.0.176_384.81_linux.run

然后按回车键或者空格键知道条款显示100%,然后如下图操作：

3、设置环境变量

打开主目录下的 .bashrc文件添加路径

sudo gedit ~/.bashrc

在末尾添加：

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

然后运行

source ~/.bashrc

4、检查是否安装成功

nvcc --version

如果显示如下，则安装成功

安装cudnn

1、下载安装文件

官网下载安装文件
我下载的是7.6.4版本
注意选择第三个下载（下载需要注册）

2、安装文件

下载完成后解压
然后执行如下命令

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、检查

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

如下则安装成功