SageMaker развертывается в EIA из режима TF Script Python3

Question

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

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

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

---

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

Answer 1

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