美文网首页
ubuntu安装cuda10.2以及cudnn7.6.5

ubuntu安装cuda10.2以及cudnn7.6.5

作者: 几夏经秋 | 来源:发表于2020-05-02 18:49 被阅读0次

    1.打开网页https://developer.nvidia.com/cuda-downloads选择目标平台

    在ubutnu 中使用wget命令下载:

    wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run

    即可完成下载:

    在完成下载之后安装cuda10.2 。

    输入命令安装cuda10.2

    sudo sh cuda_10.2.89_440.33.01_linux.run

    之后等待片刻之后会弹出安装界面:

    按照要求输入accept,之后进入选择安装组件,这里不安装驱动程序,所以驱动程序前面的x用enter键取消掉:

    使用键盘上的上下键移动光标至install,按下enter键即可等待完成安装:

    在一下弹出的窗口中按照以下光标位置来选择:

    之后等待片刻即可完成安装。

    安装好之后会有以下提示:

    安装好cuda之后开始设置环境变量。由于只是SSH连接的命令行界面。所以直接考虑使用vim或者winscp协助打开环境变量文件:

    sudo gedit /etc/profile

    原始的命令如上,可以使用winscp协助打开这个文件。打开之后在文件的最后面加入两行:

    export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}

    export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

    如下图所示:

    完成之后输入命令是其临时生效:

    source /etc/profile

    之后输入命令查看cuda及其相关组件是否安装完成

     cat /proc/driver/nvidia/version 

    nvcc -V

    两个命令输入后其结果如下:

    至此,CUDA10.2安装完成。

    完成cuda10.2之后安装cudnn7.6.5,如下:打开cudnn下载网址:

    https://developer.nvidia.com/cudnn

    之后点击下载cuDNN,这里需要你注册登录。注册登录后选择适合cuda10.2的cudnn7.6..5进行下载:

    在第三行里面的cudnn library for linux 上右击复制链接,使用ubuntu上的wget命令下载cudnn

    wget https://developer.download.nvidia.cn/compute/machine-learning/cudnn/secure/7.6.5.32/Production/10.2_20191118/cudnn-10.2-linux-x64-v7.6.5.32.tgz?RA_4nMggwWRSMbmoscOgmb1AXT0WnGpn3nqwTZSwEOLSDBoU2ABEVUO1ieZkm1Ml-C2kRfLae_Zoho3qPBCkU_yKyVLuAjFXflXTloUN2e8XLPqC0A7AhBj4G1GQ9EvbH8gSnk6xYiy2LJRqPIfSsNnpbv_vLoPUtmPAD2rFhIwvZdivT5SqnfwFi6OVpC5NZafBknkWGfL4ynFvt1L4seAhJMqSoEL4Dg

    下载完成之后

    对原文件进行重命名之后再进行解压:

    使用命令对压缩包进行解压

    tar -xvf cudnn-10.2-linux-x64-v7.6.5.32.tgz

    解压的示意图如下:

    之后执行一下命令:

    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*

    如下图:

    之后输入命令查看cudnn是否安装成功:

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

    如图所示,显示的版本是cudnn7.6.5

    即可完成CUDA+CUDNN的安装。

    相关文章

      网友评论

          本文标题:ubuntu安装cuda10.2以及cudnn7.6.5

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