• 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

  • 相关阅读:
    纯css制作带三角(兼容所有浏览器)
    xmapp 404设置
    css 选择器
    IE6/IE7/IE8下float:right的异常及其解决方法
    artdialog5 bug
    jQuery插件开发--(转)
    Global对象
    JS 关于(function( window, undefined ) {})(window)写法的理解
    wx-charts 微信小程序图表插件
    小程序二次贝塞尔曲线,购物车商品曲线飞入效果
  • 原文地址:https://www.cnblogs.com/Renyi-Fan/p/13457496.html
Copyright © 2020-2023  润新知