哪些TensorFlow和CUDA版本组合兼容? [英] Which TensorFlow and CUDA version combinations are compatible?

问题描述

我注意到某些较新的TensorFlow版本与较旧的CUDA和cuDNN版本不兼容.是否存在兼容版本的概述，甚至存在经过官方测试的组合的列表?我在TensorFlow文档中找不到它.

I have noticed that some newer TensorFlow versions are incompatible with older CUDA and cuDNN versions. Does an overview of the compatible versions or even a list of officially tested combinations exist? I can't find it in the TensorFlow documentation.

推荐答案

通常:

检查CUDA版本:

Generally:

Check the CUDA version:

cat /usr/local/cuda/version.txt

和cuDNN版本:

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

并按照下面的图片或>在此处 .

and install a combination as given below in the images or here.

以下图像和链接概述了Linux，macOS和Windows上CUDA和TensorFlow的官方支持/测试组合:

The following images and the link provide an overview of the officially supported/tested combinations of CUDA and TensorFlow on Linux, macOS and Windows:

由于在某些情况下以下给出的规格可能太宽泛，因此以下一种有效的配置有效:

Since the given specifications below in some cases might be too broad, here is one specific configuration that works:

• tensorflow-gpu==1.12.0
• cuda==9.0
• cuDNN==7.1.4

• tensorflow-gpu==1.12.0
• cuda==9.0
• cuDNN==7.1.4

可以在此处下载相应的cudnn.

The corresponding cudnn can be downloaded here.

(数据于2020年5月20日更新)

(图更新于2018年5月31日)

自2020年1月14日起更新:有关更新的信息，请参考适用于Linux的链接和适用于Windows的链接.

Updated as of 14th Jan 2020: For the updated information please refer Link for Linux and Link for Windows.

这篇关于哪些TensorFlow和CUDA版本组合兼容?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！