Pris en charge par la construction tensorflow-gpu située dans PyPI Cet article ne fait que noter les versions CUDA et cuDNN.
Je l'ajouterai à cet article après avoir confirmé l'opération avec la nouvelle version.
Cette page est créée par un individu à titre de mémo personnel tensorflow S'il vous plaît voir pour référence seulement s'il n'y a pas de description officielle.
officiel tensorflow https://www.tensorflow.org/install/source#tested_build_configurations
Windwows10
| tensorflow | CUDA | cuDNN | Remarques |
|---|---|---|---|
| 2.3.0 | 10.1 | 7.6 | Vérifiez le fonctionnement dans l'environnement de la main |
| 2.2.0 | 10.1 | 7.6 | Vérifiez le fonctionnement dans l'environnement de la main |
| 2.1.0 | 10.1 | 7.6 | Décrit dans la note de publication |
| 2.0.0 | 10.0 | 7.4 | Décrit sur le site officiel |
Ubuntu 18.04
| tensorflow | CUDA | cuDNN | Remarques |
|---|---|---|---|
| 2.3.0 | 10.1 | ||
| 2.2.0 | 10.1 | 7.6 | Vérifiez le fonctionnement dans l'environnement de la main |
| 2.1.0 | 10.1 | 7.6 | Décrit dans la note de publication |
| 2.0.0 | 10.0 | 7.4 | Décrit sur le site officiel |
La compilation tensorflow-gpu installée par pip install fait désormais référence à une version spécifique de CUDA, elle ne fonctionnera donc pas si vous incluez une version différente de CUDA.
Quelle version de tensorflow-gpu est conçue pour utiliser quelle version de CUDA ici Il n'y a pas de problème car il est répertorié dans>, mais depuis août 2020, tensorflow 2.3.0 a été publié, mais seul 2.1.0 est répertorié dans la liste. https://www.tensorflow.org/install/source#tested_build_configurations
Ensuite, s'il est répertorié dans les notes de publication, bien que la version CUDA correspondante soit écrite dans tensorflow 2.1.0, elle n'est pas répertoriée dans les notes de publication après tensorflow 2.2. https://github.com/tensorflow/tensorflow/releases
C'est un problème, donc si la nouvelle version fonctionne correctement, j'écrirai la version à ce moment-là dans cet article. J'ajouterai de nouveaux articles ici au lieu d'écrire de nouveaux articles sur la nouvelle version.
Si CUDA n'est pas inclus ou si la version n'est pas appropriée, le message d'erreur suivant s'affichera lors de l'importation. Vous pouvez indiquer la version par le nom de fichier de la DLL affichée à ce moment-là. Ci-dessous, j'essaie de lire cudart64_101.dll, donc j'ai l'impression d'avoir probablement besoin de CUDA 10.1.
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