美文网首页Tensorflow学习笔记ubuntu
Ubuntu18.04 -cuda + cudnn安装

Ubuntu18.04 -cuda + cudnn安装

作者: h小吴y | 来源:发表于2020-07-23 21:30 被阅读0次

    1.在google cloud虚拟机中安装cuda10

    在虚拟机中运行命令:

    vim installNvidia.sh

    在vim编辑器中按a键进入编辑模式,将下面一段代码复制到文件中。

    选择自己对应的cuda版本:

    https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/

    #!/bin/bash

    echo "Checking for CUDA and installing."

    # Check for CUDA and try to install.

    if ! dpkg-query -W cuda; then

      # The 18.04 installer works with 18.10.

      wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.0.130-1_amd64.deb

      sudo dpkg -i cuda-repo-ubuntu1804_10.0.130-1_amd64.deb

      sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub

      sudo apt-get update

      sudo apt-get install cuda-10.0

    fi

    echo 'export CUDA_HOME=/usr/local/cuda-10.0' >> ~/.bashrc

    echo 'export PATH=$PATH:$CUDA_HOME/bin' >> ~/.bashrc

    echo 'export LD_LIBRARY_PATH=$CUDA_HOME/lib64' >> ~/.bashrc

    source ~/.bashrc

    Esc键进入vim编辑器的命令模式,输入:wq并运行,表示保存并退出。

    在虚拟机中运行命令:

    bash installNvidia.sh

    运行少量时间后,如下图所示,询问是否继续,回答y

    经过大概10分钟,可以安装成功。

    查看Nvidia显卡的使用情况,命令:nvidia-smi

    查看CUDA版本号,命令:

    cat /usr/local/cuda/version.txt

    如果显示下图,则表示安装CUDA10.0成功

    其它安装方法:

    安装驱动

    https://blog.csdn.net/sss_369/article/details/94591280

    https://blog.csdn.net/m0_37638193/article/details/79492174

    安装多版本cuda

    解决cudnn版本不对应问题

    2.安装cudnn

    下载地址:https://blog.csdn.net/xiangxianghehe/article/details/79177833

    linux系统中下载命令为:wget {},大括号替换为下载链接。

    wget http://file.ppwwyyxx.com/nvidia/cudnn-10.0-linux-x64-v7.6.4.38.tgz 

    linux系统中解压tgz压缩文件命令为:tar -zxvf {},大括号替换为tgz压缩文件名。

    sudo tar -zxvf cudnn-10.0-linux-x64-v7.6.4.38.tgz

    解压文件后,路径下有一个cuda文件夹,运行ls命令,如下图所示:

    将cuda文件夹中的文件复制到linux的cuda程序中,两行命令如下:

    sudo cp cuda/include/cudnn.h /usr/local/cuda-10.0/include/

    sudo cp cuda/lib64/* /usr/local/cuda-10.0/lib64/

    查看cudnn版本

    在终端输入

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

    到这里为止,就已经安装cuDNN成功。

    参考链接:https://www.jianshu.com/p/893d622d1b5a

    相关文章

      网友评论

        本文标题:Ubuntu18.04 -cuda + cudnn安装

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