Ubuntu安装cuda和cudnn及测试方法

分开安装驱动和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 查看是否被禁用


Ubuntu安装cuda和cudnn及测试方法_第1张图片
没有信息输出说明禁用成功

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

        如下图则成功


Ubuntu安装cuda和cudnn及测试方法_第2张图片

(2)添加官方的ppa源

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

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

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

            sudo apt-get update  

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

            ubuntu-drivers devices


Ubuntu安装cuda和cudnn及测试方法_第3张图片
本机推荐安装415,430,390从中选择一个安装

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

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

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

        nvidia-smi


Ubuntu安装cuda和cudnn及测试方法_第4张图片

二、安装cuda和cudnn

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

(1)安装cuda

    1)验证cuda的安装条件

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


Ubuntu安装cuda和cudnn及测试方法_第5张图片

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


Ubuntu安装cuda和cudnn及测试方法_第6张图片

    最后,验证系统是否安装了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


Ubuntu安装cuda和cudnn及测试方法_第7张图片

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出现下面画面则安装成功


Ubuntu安装cuda和cudnn及测试方法_第8张图片

(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及测试方法_第9张图片

你可能感兴趣的:(Ubuntu安装cuda和cudnn及测试方法)