配置CUDA和cuDNN以及Detectron过程

Posted superfly123

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了配置CUDA和cuDNN以及Detectron过程相关的知识,希望对你有一定的参考价值。

配置CUDA和cuDNN真是心累,汗!前前后后下了n个版本,最终找到了正确的打开方式。

安装显卡

首先我的系统Ubuntu 16.04,显卡是Nvidia RTX 2060,这个显卡的驱动真心不好装,在这里也记录一下安装过程吧。
ubuntu安装显卡共有3种方式。

  • 使用标准仓库进行自动化安装
    在安装的发行版中,如 ubuntu, Linux Mint等,找到附加驱动管理软件
    技术图片

这种安装方式有如下缺点:

如果你的显卡比较新可能会出现安装低版本的NVIDIA驱动而造成即可安装完成,但是并没有真正安装成功,可能会出现循环登录,关机死机等等原因。

当你更换驱动的时候可能原有的NVIDIA驱动删除不干净。

当然这种方式也是有优点的:

不需要手动禁止nouveau
操作方便
其实ubuntu自带命令行版本安装工具ubuntu-drivers,终端输入:

ubuntu-drivers devices   # 查询所有ubuntu推荐的驱动
  • 使用PPA仓库进行自动化安装
    使用图形驱动程序PPA存储库允许我们安装NVIDIA beta驱动程序,这有可能会出现兼容性的问题,但是有些时候必须使用这种方式,比如显卡比较新,使用上面所讲的方式检测驱动的安装情况,如果不正常那么只能使用这种方式安装最新的NVIDIA驱动。
    sudo add-apt-repository ppa:graphics-drivers/ppa
    sudo apt update
    这样便可以安装成功了。
    但是我的RTX2060显卡按照这种安装方式,重启之后,屏幕一直闪烁,不能正常进入系统。无奈,进入安全模式,卸载Nvidia 驱动。
    于是我只能使用下面的这种安装方式
  • 使用官方的NVIDIA驱动进行手动安装
    下载NVIDIA官方驱动 https://www.geforce.cn/drivers sudo apt-get remove –purge nvidia*
    bios禁用禁用secure boot,也就是设置为disable 禁用nouveau
    打开编辑配置文件:
    sudo gedit /etc/modprobe.d/blacklist.conf
    在最后一行添加:blacklist nouveau
    这一条的含义是禁用nouveau第三方驱动,之后也不需要改回来。
    输入
sudo update-initramfs -u
  • 重启 reboot
  • 重启之后,可以查看nouveau有没有运行:
lsmod | grep nouveau  # 没输出代表禁用生效
  • 停止可视化桌面:

为了安装新的Nvidia驱动程序,我们需要停止当前的显示服务器。最简单的方法是使用telinit命令更改为运行级别3。执行以下linux命令后,显示服务器将停止,因此请确保在继续之前保存所有当前工作(如果有):

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

    1. 安装驱动

给驱动文件增加可执行权限:

sudo chmod a+x NVIDIA-Linux-x86_64-390.48.run
然后执行安装:

sudo sh ./NVIDIA-Linux-x86_64-390.48.run --no-opengl-files
参数介绍:

–no-opengl-files 只安装驱动文件,不安装OpenGL文件。这个参数最重要
–no-x-check 安装驱动时不检查X服务
–no-nouveau-check 安装驱动时不检查nouveau
后面两个参数可不加。
安装成功后,输入 nvidia-smi可以查看显卡信息。

安装CUDA

以上是关于配置CUDA和cuDNN以及Detectron过程的主要内容,如果未能解决你的问题,请参考以下文章

Detectron2 基准测试 | 十二

Detectron2 基准测试 | 十二

Windows下pytorch-gpu安装以及CUDA详细安装过程 附带各个版本安装包

Ubuntu16.04 安装配置 Caffe 过程 (GPU版+CUDA 9.0+cuDNN 9.0+OpenCV 3.4.1)

cuda和cudnn安装过程

安装CUDA和cuDNN