SageMaker развертывается в EIA из режима TF Script Python3 - PullRequest
0 голосов
/ 01 февраля 2019

Я установил Tensorflow Estimator в SageMaker, используя режим сценариев с framework_version='1.12.0' и python_version='py3', используя экземпляр GPU.

Вызов развертывания непосредственно в этом оценщике работает, если я выберу также тип экземпляра развертывания в качестве графического процессора.Однако если я выберу тип экземпляра ЦП и / или попытаюсь добавить ускоритель, произойдет сбой с ошибкой, из-за которой докер не сможет найти соответствующее изображение для извлечения.

Кто-нибудь знает, как обучить модель py3 на графическом процессоре в режиме сценариев, а затем развернуть на экземпляре CPU + EIA?


Я нашел частичный обходной путь, сделав промежуточный шаг создания TensorFlowModel из обучающих артефактов оценщика и затем развернув его из модели, но, похоже, это не поддерживает Python 3 (опять же, нене найти соответствующий контейнер).Если я переключусь на python_version = 'py2', он найдет контейнер, но не сможет пройти проверку работоспособности, потому что весь мой код предназначен для python 3.

1 Ответ

0 голосов
/ 11 февраля 2019

К сожалению, в настоящее время нет изображений TF + Python 3 + EI, обслуживающих изображения.Если вы хотите использовать TF + EI, вам нужно убедиться, что ваш код совместим с Python 2.

...