Unterstützt von Tensorflow-GPU-Build in PyPI In diesem Artikel werden nur die Versionen CUDA und cuDNN aufgeführt.
Ich werde es diesem Artikel hinzufügen, nachdem ich den Vorgang mit der neuen Version bestätigt habe.
Diese Seite wird von einer Person als persönliches Memo erstellt Tensorflow Nur als Referenz siehe, wenn keine offizielle Beschreibung vorliegt.
Tensorflow offiziell https://www.tensorflow.org/install/source#tested_build_configurations
Windwows10
| tensorflow | CUDA | cuDNN | Bemerkungen |
|---|---|---|---|
| 2.3.0 | 10.1 | 7.6 | Überprüfen Sie den Betrieb in der Handumgebung |
| 2.2.0 | 10.1 | 7.6 | Überprüfen Sie den Betrieb in der Handumgebung |
| 2.1.0 | 10.1 | 7.6 | Beschrieben in der Versionshinweis |
| 2.0.0 | 10.0 | 7.4 | Beschrieben auf der offiziellen Website |
Ubuntu 18.04
| tensorflow | CUDA | cuDNN | Bemerkungen |
|---|---|---|---|
| 2.3.0 | 10.1 | ||
| 2.2.0 | 10.1 | 7.6 | Überprüfen Sie den Betrieb in der Handumgebung |
| 2.1.0 | 10.1 | 7.6 | Beschrieben in der Versionshinweis |
| 2.0.0 | 10.0 | 7.4 | Beschrieben auf der offiziellen Website |
Der von pip install installierte Tensorflow-GPU-Build verweist jetzt auf eine bestimmte Version von CUDA, sodass er nicht funktioniert, wenn Sie eine andere Version von CUDA einschließen.
Welche Version von tensorflow-gpu wurde erstellt, um welche Version von CUDA hier zu verwenden Es gibt kein Problem, da es in> aufgeführt ist, aber ab August 2020 wurde Tensorflow 2.3.0 veröffentlicht, aber nur 2.1.0 ist in der Liste aufgeführt. https://www.tensorflow.org/install/source#tested_build_configurations
Wenn es dann in den Versionshinweisen aufgeführt ist, obwohl die entsprechende CUDA-Version in Tensorflow 2.1.0 geschrieben ist, wird es nach Tensorflow 2.2 nicht in den Versionshinweisen aufgeführt. https://github.com/tensorflow/tensorflow/releases
Dies ist ein Problem. Wenn die neue Version ordnungsgemäß funktioniert, werde ich die Version zu diesem Zeitpunkt in diesem Artikel aufschreiben. Ich werde hier neue Artikel hinzufügen, anstatt neue Artikel über die neue Version zu schreiben.
Wenn CUDA nicht enthalten ist oder die Version nicht geeignet ist, wird beim Importieren die folgende Fehlermeldung angezeigt. Sie können die Version anhand des Dateinamens der zu diesem Zeitpunkt angezeigten DLL erkennen. Unten versuche ich, cudart64_101.dll zu lesen, daher habe ich das Gefühl, dass ich wahrscheinlich CUDA 10.1 brauche.
python
>>> import tensorflow as tf
2020-08-05 00:33:55.037723: W tensorflow/stream_executor/platform/default/dso_loader.cc:59] Could not load dynamic library 'cudart64_101.dll'; dlerror: cudart64_101.dll not found
2020-08-05 00:33:55.037919: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
Windows10 Anaconda3 Ryzen7-1700X GTX-1080Ti
Ubuntu18.04 Anaconda3 core i9-9900K TitanV
Recommended Posts