【以下的问题经过翻译处理】 我想在 sagemaker 上提交训练任务。我在笔记本上试过,效果很好。当我尝试以下操作时,我得到ModuleNotFoundError: No module named 'nltk'
我的代码是
import sagemaker
from sagemaker.pytorch import PyTorch
JOB_PREFIX = 'pyt-ic'
FRAMEWORK_VERSION = '1.3.1'
estimator = PyTorch(entry_point='finetune-T5.py',
source_dir='../src',
train_instance_type='ml.p2.xlarge' ,
train_instance_count=1,
role=sagemaker.get_execution_role(),
framework_version=FRAMEWORK_VERSION,
debugger_hook_config=False,
py_version='py3',
base_job_name=JOB_PREFIX)
estimator.fit()
finetune-T5.py
有许多其他未安装的库。如何安装缺少的库?或者有更好的方法来运行训练任务吗?