Как получить базовый образ для Python 3.9+ при запуске учебных заданий Sagemaker в режиме сценария? ⇐ Python

[Anonymous]

Я использую Sagemaker в AWS Sagemaker Studio, чтобы попытаться выполнить задания обучения в режиме сценария.
Ниже приведен пример фрагмента кода, который я пытаюсь запустить:

Код: Выделить всё

from sagemaker.xgboost.estimator import XGBoost
est = XGBoost(
entry_point=path_to_script,
instance_type=instance_type,
framework_version="1.7-1", #latest
role=exec_role,
instance_count=1,
dependencies=[path_to_requirements],
sagemaker_session=sm_session,
)
est.fit()

Сценарий path_to_script написан на Python 3.9+.
Задание обучения завершается сбоем во время создания образа, поскольку кажется, что на изображении XGBoost работает Python 3.8. Как мне выбрать изображение, соответствующее версии 3.9+?
Примечание: сценарий, который я запускаю, на самом деле не зависит от XGBoost. Я использую его для запуска собственного сценария обучения, следуя примеру режима сценария, который можно найти здесь: https://sagemaker-examples.readthedocs. ... -mode.html
Я пробовал разные версии XGBoost, последняя поддерживает только Python 3.8
Я также поискал доступные изображения в Sagemaker. Я вижу, что у Sagemaker есть разные обучающие изображения, которые можно найти здесь: https://docs.aws.amazon.com/sagemaker/l ... mages.html
но это мне не ясно, как выбрать одного из них для запуска учебного задания.

Подробнее здесь: https://stackoverflow.com/questions/784 ... ng-jobs-in