zoukankan      html  css  js  c++  java
  • TensorFlow1.2~2.1各个GPU版本CUDA和cuDNN对应版本整理(干货)

    TensorFlow1.2~2.1各个GPU版本CUDA和cuDNN对应版本整理(干货)

    一、总结

    一句话总结:

    可以在官网查看tensorflow、cuda、cudnn的对应关系,在TensorFlow的GPU版本安装过程中,如果对应的CUDA版本和cuDNN版本不正确的话,是无法正常使用GPU来进行模型训练的。

    二、干货:TensorFlow1.2~2.1各个GPU版本CUDA和cuDNN对应版本整理

    转自或参考:干货:TensorFlow1.2~2.1各个GPU版本CUDA和cuDNN对应版本整理
    https://blog.csdn.net/fengdu78/article/details/104667044

    最近发现很多QQ群和微信群里经常会有人问这么一个问题——“我安装TensorFlow GPU版本怎么总是提示CUDA版本不对或者cuDNN版本不对呢?”为了解决大家这个问题,我特意把TensorFlow1.2到最新版本所需要的CUDA和cuDNN对应的版本做了个整理,希望能够对大家有帮助。

    要搭建TensorFlow的GPU版本,首先需要的必备条件就是一块能够支持CUDA的NVIDIA显卡,因为在搭建TensorFlow的GPU版本时,首先需要做的一件事就是安装其基础支持平台CUDA和其机器学习库cuDNN,然后在此基础上搭建TensorFlow GPU版本。

    其次还要了解一下不同的TensorFlow版本所需要对应安装的CUDA和cuDNN版本是多少,因为在TensorFlow的GPU版本安装过程中,如果对应的CUDA版本和cuDNN版本不正确的话,是无法正常使用GPU来进行模型训练的。下表整理出了TensorFlow从1.2到最新版本的CUDA和cuDNN所对应的版本集合。

    TensorFlow版本

    CUDA版本

    cuDNN版本

    1.2

    CUDA Toolkit 8.0

    cuDNN v5.1

    1.3

    CUDA Toolkit 8.0

    cuDNN v6 or v6.1

    1.4

    CUDA Toolkit 8.0

    cuDNN v6.1

    1.5

    CUDA Toolkit 9.0

    cuDNN v7.0

    1.6

    CUDA Toolkit 9.0

    cuDNN v7.0

    1.7

    CUDA Toolkit 9.0

    cuDNN v7.0

    1.8

    CUDA Toolkit 9.0

    cuDNN v7.0

    1.9

    CUDA Toolkit 9.0

    cuDNN v7.0

    1.10

    CUDA Toolkit 9.0

    cuDNN v7.1

    1.11

    CUDA Toolkit 9.0

    cuDNN v7.1

    1.12

    CUDA Toolkit 9.0

    cuDNN v7.3

    1.13

    CUDA Toolkit 10.0

    cuDNN v7.3

    1.14

    CUDA Toolkit 10.0

    cuDNN v7.4

    1.15

    CUDA Toolkit 10.0

    cuDNN v7.5

    2.0

    CUDA Toolkit 10.0

    cuDNN v7.6

    2.1.0

    CUDA Toolkit 10.1

    cuDNN v7.6


     

    这个页面可以查看这三者版本的对应关系

    https://tensorflow.google.cn/install/source_windows

  • 相关阅读:
    L2-004. 这是二叉搜索树吗?
    CF934A A Compatible Pair
    CF937B Vile Grasshoppers
    CF940B Our Tanya is Crying Out Loud
    ZOJ 3182 Nine Interlinks
    ZOJ 3175 Number of Containers
    Codeforces Round #193 (Div. 2) B
    CodeForces 149D Coloring Brackets
    POJ 2653 Pick-up sticks(计算几何)
    UVA 12506 Shortest Names
  • 原文地址:https://www.cnblogs.com/Renyi-Fan/p/13457496.html
Copyright © 2011-2022 走看看