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

问题描述

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

解决方案

TL;DR) 请参阅此表:

Linux CPU

macOS GPU

macOS CPU

Windows GPU

Windows CPU

截至 2020 年 12 月 5 日更新:有关更新的信息，请参阅 Linux 链接 和 Windows 链接.

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.

解决方案

TL;DR) See this table: https://www.tensorflow.org/install/source#gpu

Generally:

Check the CUDA version:

cat /usr/local/cuda/version.txt

and cuDNN version:

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

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

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

Minor configurations:

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

The corresponding cudnn can be downloaded here.

Tested build configurations

Please refer to https://www.tensorflow.org/install/source#gpu for a up-to-date compatibility chart (for official TF wheels).

(figures updated May 20, 2020)

Linux GPU

Linux CPU

macOS GPU

macOS CPU

Windows GPU

Windows CPU

Updated as of Dec 5 2020: For the updated information please refer Link for Linux and Link for Windows.

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