美文网首页
Ubuntu16.4安装CUDA9.0+CUDNN7.1

Ubuntu16.4安装CUDA9.0+CUDNN7.1

作者: 小小虫_56d2 | 来源:发表于2019-03-06 20:27 被阅读0次

    按照上一篇文章的显卡驱动安装方法完成后,进行CUDA和CUDNN的安装,注意不建议更新系统的python版本,如果升级到python3.6的话后边会很多莫名其妙的错误,如终端打不开、系统更新打不开、显卡切换工具打不开等等,需要python3.6的话建议在Anaconda中配置。

    建议严格按照步骤执行,不然可能会有意想不到的问题

    ~~~~~~~~~~~~~~~~~分割线~~~~~~~~~~~~~~~~~

    1 准备工作,下载CUDA和CUDNN,

    CUDA:https://developer.nvidia.com/cuda-toolkit-archive

    CUDNN:https://developer.nvidia.com/rdp/cudnn-archive

    我这里是cuda9.0 cudnn7.1。下载CUDNN时要对应好CUDA的试用版本!建议不要选择最新的版本。

    CUDA选择:runfile安装包

    image

    CUDNN选择:压缩包安装方法 cuDNN v7.1.4 Library for linux

    image

    2 安装准备

    查看gcc是否安装及版本,

    在终端中输入:

    gcc --version
    
    image

    gcc版本太低的话会不支持,升级方法自行百度

    3 安装CUDA

    终端中转到下载的目录,输入:

    sudo sh cuda_9.0.176_384.81_linux.run
    

    会有很长的协议,空格可退出。最后才进入正式安装,根据下图输入,注意第二步中安装显卡驱动一定要选no,因为之前已经装过了,不然会出很多莫名其妙的问题。其他都是yes

    image

    安装完成后会如下显示,最后的warning不用管他

    image

    完成后需要设置环境变量,在home目录下按ctrl+h显示隐藏文件,打开.bashrc

    image

    在最后输入:

    exportLD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
    
    exportPATH=$PATH:/usr/local/cuda-9.0/bin  #其中cuda-9.0根据你的cuda版本修改
    
    image

    保存,然后在终端更新环境变量:

    source ~/.bashrc
    

    完成,验证是否安装成功:

    cd NVIDIA_CUDA-9.0_Samples/1_Utilities/deviceQuery
    
    make
    
    ./deviceQuery
    

    过程及结果如下:

    image

    或者输入:

    nvcc --version
    
    image

    4 安装cudnn

    终端转到cudnn下载位置,输入:

    sudo tar -zxvf ./cudnn-9.0-linux-x64-v7.1.tgz
    

    解压完成后有两个文件夹include和lib64,然后将/include目录下的文件拷贝到/usr/local/cuda/include中,如下命令:

    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*
    

    查看是否安装成功:

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

    然后cudnn就配置完成了!

    image

    相关文章

      网友评论

          本文标题:Ubuntu16.4安装CUDA9.0+CUDNN7.1

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