美文网首页
Ubuntu安装cuda和cudnn及测试方法

Ubuntu安装cuda和cudnn及测试方法

作者: 张亦风 | 来源:发表于2019-08-01 21:06 被阅读0次

分开安装驱动和cuda

1、安装显卡驱动

(1)方法一从官网下载驱动编译

        NVIDIA的中文官网为nvidia显卡驱动

1)打开终端输入以下命令删除旧驱动

        sudo apt-get purge nvidia*

2)禁用自带的 nouveau nvidia驱动

        首先利用命令行打开

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

        在文件末端加入

        blacklist nouveau  

        options nouveau modeset=0

        然后更新

        sudo update-initramfs -u

        最后重启

        终端下输入命令 lsmod | grep nouveau 查看是否被禁用

没有信息输出说明禁用成功

3)下载好对应gpu版本.run 形式的显卡驱动

         图形化界面按住ctrl+alt+(f1~f6的其中一个)进入命令行模式输入用户名和密码,然后在输入

         sudo service lightdm stop 

4)给驱动run文件赋予执行权限:

         sudo chmod  a+x NVIDIA-Linux-x86_64-396.18.run

         然后开始安装

         sudo ./NVIDIA-Linux-x86_64-396.18.run 

5)挂载显卡驱动

        modprobe nvidia

6)检查是否安装成功

         nvidia-smi

        如下图则成功

(2)添加官方的ppa源

   1)快捷键ctrl+alt+T打开命令终端输入

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

   2)然后刷新软件库并安装最新驱动

            sudo apt-get update  

   3)使用终端命令查看Ubuntu推荐的驱动版本

            ubuntu-drivers devices

本机推荐安装415,430,390从中选择一个安装

4)采用apt-get命令在终端安装

        sudo apt-get install nvidia-430 nvidia-settings nvidia-prime

5)重启系统并验证  打开终端并输入

        nvidia-smi

二、安装cuda和cudnn

cuda的官方网址cudnn的官方网址,cudnn的下载需要注册账号。

(1)安装cuda

    1)验证cuda的安装条件

    首先,在终端输入 lspci | grep -I nvidia 查看你的gpu版本信息

      接着,验证是否安装gcc,终端输入 gcc –version ,如果未安装,则输入sudo apt-get  install  build-essential进行安装

    最后,验证系统是否安装了kernel header和 package development

    终端输入 uname –r 查看内核 接着输入sudo apt-get install linux-headers-$(uname -r)

    即   sudo apt-get install linux-headers-4.18.20-25-generic

    就可以安装对应kernel版本的kernel header和package development

2)安装cuda

    终端输入 sudo sh 你下载的cuda.run

    然后一直按住回车键,直到服务条款显示到100%。除了driver选项选择no其他都选择yes。

3)设置环境变量

    在终端输入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

    目录要对应自己的cuda版本

4)然后source  ~/.bashrc 激活环境

5)在终端输入nvcc -V出现下面画面则安装成功

(2)安装cudnn

    1)下载对应你cuda版本的cudnn

    2)安装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*

3)查看cudnn版本

    在终端输入

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

    出现以下信息则表明安装成功

相关文章

网友评论

      本文标题:Ubuntu安装cuda和cudnn及测试方法

      本文链接:https://www.haomeiwen.com/subject/vdpcdctx.html