Я пытаюсь обучить модель pytorch на студии amazon sagemaker.
Это работает, когда я использую EC2 для обучения с:
estimator = PyTorch(entry_point='train_script.py',
role=role,
sagemaker_session = sess,
train_instance_count=1,
train_instance_type='ml.c5.xlarge',
framework_version='1.4.0',
source_dir='.',
git_config=git_config,
)
estimator.fit({'stockdata': data_path})
и работает в локальном режиме в классе c блокнот sagemaker (не студия) с:
estimator = PyTorch(entry_point='train_script.py',
role=role,
train_instance_count=1,
train_instance_type='local',
framework_version='1.4.0',
source_dir='.',
git_config=git_config,
)
estimator.fit({'stockdata': data_path})
Но когда я использую тот же код (с train_instance_type = 'local') в студии sagemaker, он не работает, и у меня возникает следующая ошибка: Нет такого файла или каталога: 'docker': 'docker'
Я пытался установить docker с помощью установки pip, но команда docker не найдена, если использовать его в терминале