J'ai installé le Tensorflow r1.14 et je souhaite utiliser TF-TRT . Cependant, l'erreur suivante se produit:
"ModuleNotFoundError: Aucun module nommé 'tensorflow.contrib.tensorrt'"
lors de l'exécution de l'exemple de code. La même erreur se produit avec Tensorflow r1.13 . Ma question est donc la suivante: dois-je installer la bibliothèque tensorflow.contrib.tensorrt séparément? Si oui, comment?
De plus, je peux exécuter l'exemple de code du TensorRT , par exemple sampleINT8 , avec succès. Cliquez ici pour voir mon exemple de code exécuté avec succès.
Cela me porte à croire que TensorRT est correctement installé. Cependant, le TF-TRT ne fonctionne toujours pas.
Toute aide serait grandement appréciée!
4 Réponses :
Pour pouvoir importer tensorflow.contrib.tensorrt , vous devez avoir tensorflow-gpu version> = 1.7 installé sur votre système . Vous pourriez peut-être essayer d'installer la bibliothèque tensorflow-gpu avec un:
pip install tensorflow-gpu
Consultez également la section Windows de la documentation GPU . Aussi, j'essaierais de mettre à jour votre version de tensorflow avec un:
pip install --upgrade tensorflow
pour vous assurer que vous y êtes également à jour. Consultez cette section de la documentation TensorFlow pour une assistance supplémentaire.
J'espère que cela vous aidera!
La tf .__ version__ est 1.13.1 ce qui est> = 1.7. J'ai également vérifié le dossier «tensorflow \ contrib» et il n'y a pas de sous-dossier appelé tensorrt. Je ne pense pas que l'installation de pip propose une tension. Que puis-je faire d'autre maintenant?
Vous pourriez peut-être essayer de télécharger le pilote tensorrt directement depuis nvidia: docs.nvidia.com/deeplearning/sdk/tensorrt-install-guide/...
Dans TF 1.14, TF-TRT a été déplacé vers le noyau depuis contrib.
Vous devez l'importer comme ceci:
de tensorflow.python.compiler.tensorrt import trt_convert comme trt
2 possibilités
tensorrt sous contrib dans la distribution de Windows TF, mais Linux l'a (j'ai essayé 1.13.1). Dans TF 1.14, TF-TRT a été déplacé vers le noyau depuis contrib.
Vous devez l'importer comme ceci: depuis tensorflow.python.compiler.tensorrt import> trt_convert as trt
C'est la bonne réponse pour Linux.
Cependant, si vous utilisez Windows: l'API TensorRT Python (et donc TF-TRT) n'est pas prise en charge pour Windows pour le moment, donc les packages python TensorFlow ne sont pas construits avec TensorRT.