it-swarm.com.de

Welche TensorFlow- und CUDA-Versionskombinationen sind kompatibel?

Ich habe festgestellt, dass einige neuere TensorFlow-Versionen nicht mit älteren CUDA- und cuDNN-Versionen kompatibel sind. Gibt es eine Übersicht der kompatiblen Versionen oder sogar eine Liste von offiziell getesteten Kombinationen? Ich kann es in der TensorFlow-Dokumentation nicht finden.

70
Fábio

Allgemein:

Überprüfen Sie die CUDA-Version:

cat /usr/local/cuda/version.txt

und cuDNN Version:

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

und installiere eine Kombination wie unten in den Bildern angegeben oder hier .

Die folgenden Bilder und der Link bieten einen Überblick über die offiziell unterstützten/getesteten Kombinationen von CUDA und TensorFlow unter Linux, macOS und Windows:

Kleinere Konfigurationen:

Da die unten angegebenen Spezifikationen in einigen Fällen zu weit gefasst sein können, funktioniert eine bestimmte Konfiguration:

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

Der entsprechende cudnn kann heruntergeladen werden hier .

(Zahlen aktualisiert am 29. Juni 2019)

Linux-GPU

enter image description here

Linux

enter image description here

macOS GPU

enter image description here

mac OS

enter image description here

(Abbildung aktualisiert am 31. Mai 2018)

Windows

enter image description here

105
Fábio

Die Kompatibilitätstabelle in https://www.tensorflow.org/install/source#tested_build_configurations enthält keine speziellen Nebenversionen für cuda und cuDNN. Es wird nur allgemein als cuda = 9 und cuDNN = 7 aufgeführt. Wenn die spezifischen Versionen jedoch nicht erfüllt sind, tritt ein Fehler auf.

Für tensorflow-gpu==1.12.0 und cuda==9.0 lautet die kompatible cuDNN Version 7.1.4, die nach der Registrierung unter hier heruntergeladen werden kann.

Sie können Ihre cuda-Version mit überprüfen
nvcc --version

cuDNN-Version mit
cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2

tensorflow-gpu version mit
pip freeze | grep tensorflow-gpu

12
Atul Balaji

Arbeiten: Tensorflow 1.13.1, CUDA 10, CUDNN 7.4.2, python 3.6 (funktioniert nicht gut mit 3.7 .. 3.7 hat viele Fehler) Für Windows 10

4
Ketki Shroff

Du kannst diese Konfiguration für cuda 10.0 verwenden (10.1 funktioniert nicht ab 3/18), dies läuft für mich:

  • tensorflow> = 1,12,0
  • tensorflow_gpu> = 1,4

Installieren Sie die Version tensorflow gpu:

pip install tensorflow-gpu==1.4.0
1