使用GPU功能在tensorflow-hub上使用retrain.py重新训练图像

Question

我是Tensorflow的新手，我正在使用retrain.py训练一些图像。如果我有一个较大的10000张图像的数据库，并且我具有支持GPU的系统。我如何使用retrain.py在Nvidia GPU上运行。这样可以更快地进行培训。

我正在按照下面的链接进行操作

https://www.tensorflow.org/hub/tutorials/image_retraining

Answer 1

要获得GPU支持，请确保安装PIP软件包tensorflow-gpu而不是普通tensorflow。您应该从retrain.py那里看到一些性能优势。也就是说，retrain.py可以显示其年龄（早于TF Hub），并且不能很好地利用GPU，因为它在提取瓶颈值时无法正确地批处理图像。

如果您准备好住在TF 2.0.0alpha0（上周announced）的最前沿，请看一下Hub的 examples/colab/tf2_image_retraining.ipynb，它体积更小，速度更快（如果使用GPU，则 ），甚至还支持微调图像模块。