美文网首页
Ubuntu18.04安装CUDA/CUDNN

Ubuntu18.04安装CUDA/CUDNN

作者: AntiGravity | 来源:发表于2020-05-13 00:14 被阅读0次

    硬件:RTX2080

    1. 安装驱动

    sudo apt-get install nvidia-driver-440 # 440是驱动版本号,输到前一位按tab后选择最新
    

    测试:

    nvidia-msi # 显示驱动版本440和驱动的CUDA版本10.2(和运行CUDA不同)
    

    2. 安装CUDA

    目前,为适配TensorFlow1.15建议使用CUDA 10.0版本。
    TensorFlow各版本对CPU GPU版本的需求:官网
    有一种说法,安装1.15的tensorflow-gpu后,自动安装相应的cuda和cudnn,不过我没有尝试过。

    pip3 install tensorflow-gpu==1.15
    

    3. 安装CUDNN

    官网下载runtime/dev/doc三个版本的deb文件,依次安装。
    官网说明

    • 安装后查看版本:
    cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2
    
    • 用sample验证CUDNN(需要之前安装dev和doc两个deb)

    修改cudnn.h的include "..."为include <...>,否则下面编译会报错

    cd  /usr/src/cudnn_samples_v7/mnistCUDNN # 如果没有操作权限就cp到可操作的位置
    make clean && make
    ./minstCUDNN
    

    相关文章

      网友评论

          本文标题:Ubuntu18.04安装CUDA/CUDNN

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