tensorflow - 哪些 TensorFlow 和 CUDA 版本组合兼容？

Question

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

Answer 1

TL;DR) 请参阅此表:https://www.tensorflow.org/install/source#gpu

一般:

检查CUDA版本:

cat /usr/local/cuda/version.txt

和 cuDNN 版本:

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

并安装如下图像中给出的组合或 here .

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

次要配置:

由于下面给出的规范在某些情况下可能过于宽泛，因此以下是一种有效的特定配置:

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

可以下载对应的cudnn here .

经过测试的构建配置

请引用https://www.tensorflow.org/install/source#gpu获取最新的兼容性图表(针对官方 TF 车轮)。

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

Linux GPU

Linux CPU

macOS GPU

macOS CPU

Windows GPU

Windows CPU

更新截至2020年12月5日:更新信息请参阅Link for Linux和 Link for Windows .