在 Sagemaker 上开始训练任务时出现 ModuleNotFoundError

【以下的问题经过翻译处理】 我想在 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 有许多其他未安装的库。如何安装缺少的库？或者有更好的方法来运行训练任务吗？