微信扫一扫打赏支持

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

 

 

 

 

 

 

posted @ 2020-08-08 18:47  范仁义  阅读(3238)  评论(0编辑  收藏  举报