• Linux从入门到适应(四):Ubuntu 16.04环境下,安装Nvidia驱动,cuda9.2和 cudnn


    在安装深度学习框架之前,cuda和cudnn是必须要提前安装的,现在按照流程而nvidia驱动的版本和cuda版本有这一些对应关系,所以需要按照版本进行安装,现在说一下如何安装:

    1 安装nvidia驱动:

    (1)卸载旧版本的显卡驱动:

    sudo apt-get purge nvidia*

    (2)添加Graphic Drivers PPA

    sudo add-apt-repository ppa:graphics-drivers/ppa

    sudo apt-get update

    (3)去英伟达官网,或者使用 ubuntu-drivers devices查看最佳版本的驱动好,下载驱动。驱动选择.run文件,并且复制移动到一个方便 cd 的英文文件夹下面,比如:/home/。

    (4)禁用nouveau第三方驱动,

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

            在最后一行增加 blacklist nouveau

            改好后执行命令:sudo update-initramfs -u

            重启:reboot

    (5)按Ctrl+Alt+F1进入命令行界面      

             执行命令:lsmod | grep nouveau

             禁用X服务:sudo /etc/init.d/lightdm stop

             进入你存放驱动.run文件的文件夹:cd /home

             给驱动run文件赋予可执行权限:sudo chmod a+x  NVIDIA-Linux-x86_64-410.78.run

             安装 sudo ./NVIDIA-Linux-x86_64-410.78.run -no-opengl-files  安装过程中按照提示直接的默认值直接按

             开启X服务:sudo /etc/init.d/lightdm start   

             重启,用nvidia-smi检查  

    2 安装cuda

    (1)cuda文件去nvidia官网下载,可以选择.run或者.deb文件都可以,版本与你的显卡要对应,以及深度学习的框架是否支持,不建议下载最新版本的。

    (2)按照官网下面的提示命令安装。

    (3)在~/.bashrc 中设置环境变量:

    exportPATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}} 
    exportLD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
    

      运行  source ~/.bashrc 重启。

    (4)检测cuda安装是否成功,如果是默认安装,则cuda的测试sample在/usr/local/cuda-9.2/samples/1_Utilities/deviceQuery (我的cuda版本为9.2)

             直接运行 makefile文件:sudo make

             执行生成的可执行文件:./deviceQuery

             如何结果为  deviceQuery, CUDA Driver = CUDART, CUDA Driver Version = 9.2, CUDA Runtime Version = 9.2, NumDevs = 1   Result = PASS  说明安装成功

    3 安装cudnn

    (1)去官网登录,下载cudnn文件,选择与cuda版本对应的cudnn文件,接着选择 cuDNN Library for Linux

    (2)解压缩: tar -zxvf cudnn-9.2-linux-x64-v7.4.1.5.tgz

    (3)将相应的头文件复制到cuda-9.2文件夹下对应的位置:

    sudo cp cuda/include/cudnn.h /usr/local/cuda-9.2/include
    sudo cp cuda/lib64/libcudnn* /usr/local/cuda-9.2/lib64
    

    (4)修改文件权限:

    sudo chmod a+r /usr/local/cuda-9.2/include/cudnn.h /usr/local/cuda-9.2/lib64/libcudnn*
  • 相关阅读:
    简述智障版本搜索引擎架构
    kaggle PredictingRedHatBusinessValue 简单的xgboost的交叉验证
    机器学习速查表
    World final 2017 题解
    微博爬虫
    喵哈哈村的魔法考试 Round #21 (Div.2) 题解
    喵哈哈村的魔法考试 Round #20 (Div.2) 题解
    Tinkoff Challenge
    常用的机器学习&数据挖掘知识(点)总结
    喵哈哈村的魔法考试 Round #19 (Div.2) 题解
  • 原文地址:https://www.cnblogs.com/code-wangjun/p/9986255.html
Copyright © 2020-2023  润新知