美文网首页
Ubuntu安装CUDA、CUDNN

Ubuntu安装CUDA、CUDNN

作者: 糊糊小鱼干 | 来源:发表于2018-06-01 21:36 被阅读0次

    适用于多用户服务器安装于自己目录的情况

    因为安装好Anaconda之后其实已经自带了CUDA、CUDNN这些,但是我发现实际配置caffe还是会有各种各样的问题,所以重新自己安装了

    CUDA

    首先在官网下载需要的CUDA版本的安装包,以CUDA9.0版本为例,选择后会引导一步步地选择合适的版本,我的选择如下图:

    1

    即可开始下载cuda_9.0.176_384.81_linux.run

    下载好后,安装CUDA

    chmod +x cuda_9.0.176_384.81_linux.run
    ./cuda_9.0.176_384.81_linux.run
    

    添加环境变量

    vi  ~/.bashrc
    export PATH="/yourpath/cuda-9.0/bin:$PATH"
    export LD_LIBRARY_PATH=/yourpath/cuda-9.0/lib64
    
    source ~/.bashrc
    

    查看自己现在安装的CUDA版本

    nvcc -V
    

    问题

    找不到nvcc

    原因是环境路径未设置,设置PATH和LD_LIBRARY_PATH

    export PATH="/yourpath/cuda-9.0/bin:$PATH"
    export LD_LIBRARY_PATH=/yourpath/cuda-9.0/lib64
    

    cuDNN

    需要根据上一步安装的CUDA版本来下载对应的cuDNN安装包,这里我下载了cudnn-9.0-linux-x64-v7.1.tgz

    解压文件

    tar xvf cudnn-9.0-linux-x64-v7.1.tgz
    

    得到一个cuda文件夹

    cd cuda
    cp include/cudnn.h /yourpath/cuda-9.0/include/
    cp lib64/libcudnn* /yourpath/cuda-9.0/lib64/
    chmod a+r /yourpath/cuda-9.0/include/cudnn.h /yourpath/cuda-9.0/lib64/libcudnn*
    

    查看cuDNN版本

    cat /yourpath/cuda-9.0/include/cudnn.h |grep CUDNN_MAJOR -A 2
    

    输出

    #define CUDNN_MAJOR 7
    #define CUDNN_MINOR 1
    #define CUDNN_PATCHLEVEL 1
    --
    #define CUDNN_VERSION    (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
    
    #include "driver_types.h"
    

    可知cuDNN版本为7.1.1

    相关文章

      网友评论

          本文标题:Ubuntu安装CUDA、CUDNN

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