Я установил Tensorflow Estimator в SageMaker, используя режим сценария с framework_version='1.12.0'
и python_version='py3'
, используя экземпляр GPU.
Вызов развертывания непосредственно в этом оценщике работает, если я также выберу тип экземпляра развертывания как GPU. Однако, если я выберу тип экземпляра ЦП и / или попытаюсь добавить ускоритель, он выйдет из строя с ошибкой, что докер не может найти соответствующее изображение для извлечения.
Кто-нибудь знает, как обучить модель py3 на графическом процессоре в режиме сценария, а затем развернуть на экземпляре CPU + EIA?
Я нашел частичный обходной путь, выполнив промежуточный шаг создания TensorFlowModel из обучающих артефактов оценщика и затем развернув его из модели, но это, похоже, не поддерживает python 3 (опять же, не находит соответствующий контейнер). Если я переключусь на python_version = 'py2', он найдет контейнер, но не пройдет проверку работоспособности, потому что весь мой код предназначен для python 3.