ubuntu16.04 RTX2060||安装nvidia显卡驱动|cuda|cudnn

元学习论文总结||小样本学习论文总结

2017-2019年计算机视觉顶会文章收录 AAAI2017-2019 CVPR2017-2019 ECCV2018 ICCV2017-2019 ICLR2017-2019 NIPS2017-2019

安装显卡驱动

一、查看适配显卡版本并下载

到英伟达官网搜索驱动 

我的显卡是RTX2060

二、卸载自带驱动

1.卸载原先的驱动

sudo apt-get remove --purge nvidia*

2.禁用原有驱动nouveau（有两种方法，我把两种方法都操作了，才成功禁用）

法一：

cd /lib/modules/4.0.0-12-generic/kernel/drivers/gpu/drm/nouveau
# 注意：上面命令中的 4.0.0-12-generic 根据ubuntu版本的不同而不同，具体是哪一个可以自己查看
# 如果文件夹下有两个不同版本4.0.0-**-generic，都执行如下操作

mv nouveau.ko nouveau.ko.org

更新：

update-initramfs -u

查看是否禁用成功，如果无结果显示则表明成功禁用：

lsmod | grep nouveau

有结果，继续方法二

法二：

sudo vim /etc/modprobe.d/blacklist-nouveau.conf

# 在最后添加下列两行

blacklist nouveau

options nouveau modeset=0


# 更新重新生成 kernel initramfs

update-initramfs –u

查看是否禁用成功：

lsmod | grep nouveau

题主到这里没有再打印信息，说明禁用成功(不用重启)

 

三、开始安装驱动

进入ubuntu后，在桌面按下Ctrl+Alt+F1进入命令行界面，输入用户名密码登陆
执行命令

# 关闭图形界面
sudo service lightdm stop

#进入去驱动程序所在的默认下载目录，上面的username为你的用户名，如果你把放在其他地方，进入相应的目录
cd /home/username/Downloads

#后面的驱动要改为你自己的驱动，可在输入NVIDIA后按下Tab键自动补全
sudo chmod a+x NVIDIA-Linux-x86_64-390.77.run

sudo ./NVIDIA-Linux-x86_64-390.77.run -no-opengl-files

对于所有选项，都选了yes

待安装完成后,打开图像界面

sudo service lightdm start

进入系统后在终端输入

nvidia-smi
如果可以查看显卡的信息，表明安装成功，如果不行，尝试重启后再输入命令查看。

 安装cuda

建议安装9.0或9.1版本（而不必与显卡适配的版本10.0，因为应用支持不好），这里安装9.1

1.下载文件
官网地址：https://developer.nvidia.com/cuda-zone 选择适合自己的版本

2.安装

sudo chmod a+x cuda_*.0.130_410.48_linux.run // 获取权限，用tab扩充

sudo sh cuda_*.0.130_410.48_linux.run 

这时会出现很长的声明，一直按enter键，把声明读完，然后就可以进行选择了。

对于第一个选项install nvidia accelerated Graphics Driver，选n (为了避免内核编译错误)（这会导致安装完成有一个警告，可以忽略）

安装完成后会出现个提醒，这是因为我在安装CUDA的时候没有选择安装驱动，提示需要安装驱动，忽略就行。

3.加入环境路径
打开.bashrc 文件：sudo gedit ~/.bashrc
打开文件后将下面两句话加入进去：

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

保存 关闭，命令行输入：source ~/.bashrc

4.终端输入：nvcc --version 会输出CUDA的版本信息

这样就是安装成功了！

 

安装cudnn

1.官网下载：https://developer.nvidia.com/rdp/cudnn-download
进入时需要登录，没有账户的话就注册一个，进入即可。选择和自己cuda适配的版本。9.1对应cudnn7.1
然后下载，是一个压缩包，我下载完之后电脑自动跳出来提取，是一个cuda文件夹，然后我就提取出来了，放到了home文件夹下。如果不能直接提取的，百度搜Ubuntu下解压的命令，将它解压到一个确定的文件夹下。
2.右击cuda文件夹，选择在终端打开，然后输入

sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp 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版本

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