Я новичок в Tensorflow, я использую retrain.py для обучения некоторых изображений. В случае, если у меня есть большая база данных из 10000 изображений, и у меня есть система с поддержкой графического процессора. Как я могу использовать retrain.py для работы на моем графическом процессоре Nvidia. Так что обучение будет проходить быстрее.
Я следую шагам по ссылке ниже
Чтобы получить поддержку графического процессора, обязательно установите пакет PIP tensorflow-gpu вместо обычного tenorflow. Вы должны увидеть некоторые преимущества в производительности для retrain.py. Тем не менее, retrain.py показывает свой возраст (намного раньше, чем TF Hub) и не так хорошо использует графические процессоры, потому что он не выполняет правильную пакетную обработку изображений при извлечении значений узких мест.
Если вы готовы жить на переднем крае TF 2.0.0alpha0 (объявлено последним, неделю), взгляните на Hub examples / colab / tf2_image_retraining.ipynb, который значительно меньше, быстрее (если вы используете графический процессор) и даже поддерживает тонкую настройку модуля изображения.
