Какие комбинации версий TensorFlow и CUDA совместимы?

Question

Я заметил, что некоторые новые версии TensorFlow несовместимы со старыми версиями CUDA и cuDNN.Существует ли обзор совместимых версий или даже список официально протестированных комбинаций?Я не могу найти это в документации TensorFlow.

Answer 1

Работает: tenenflow 1.13.1, CUDA 10, CUDNN 7.4.2, python 3.6 (плохо работает с 3.7 .. 3.7 имеет много ошибок) Для Windows 10

Answer 2

Таблица совместимости, приведенная в https://www.tensorflow.org/install/source#tested_build_configurations, не содержит специальных вспомогательных версий для cuda и cuDNN.Он обычно указывается как cuda = 9 и cuDNN = 7.Однако, если конкретные версии не соблюдаются, возникнет ошибка.

Для tensorflow-gpu==1.12.0 и cuda==9.0, совместимая версия cuDNN - 7.1.4, которую можно загрузить с здесь после регистрации.

Вы можете проверить свою версию cuda, используя
nvcc --version

cuDNN, используя
cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2

tenorflow-gpuверсия с использованием
pip freeze | grep tensorflow-gpu

Answer 3

Вы можете использовать эту конфигурацию для cuda 10.0 (10.1 не работает с 3/18), это работает для меня:

• тензор потока> = 1.12.0
• tenorsflow_gpu> = 1.4

Установить версию tenorflow gpu:

pip install tensorflow-gpu==1.4.0

Answer 4

Обычно:

Проверьте версию CUDA:

cat /usr/local/cuda/version.txt

и версию cuDNN:

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

и установите комбинацию, как показано ниже на изображениях или здесь .

Следующие изображения и ссылка предоставляют обзор официально поддерживаемых / протестированных комбинаций CUDA и TensorFlow в Linux, macOS и Windows:

Незначительные конфигурации:

Поскольку приведенные ниже спецификации в некоторых случаях могут быть слишком широкими, вот одна конкретная конфигурация, которая работает:

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

Соответствующий cudnn можно загрузить здесь .

(цифры обновлены 29 июня,2019)

Linux GPU

Linux

macOS GPU

macOS

(цифра обновлена ​​31 мая 2018 г.)

Ветервл.