ImportError：在 SageMaker 中运行批量转换作业时没有名为 cv2 的模块答案

【问题标题】：ImportError: No module named cv2 when run Batch transform jobs in SageMakerImportError：在 SageMaker 中运行批量转换作业时没有名为 cv2 的模块
【发布时间】：2020-04-11 04:51:46
【问题描述】：

当我尝试在 AWS SageMaker 中运行批量转换作业时，遇到以下错误：

ImportError: 没有名为 cv2 的模块

请注意，我可以在笔记本实例中“导入 CV2”。 jupter 可以在 notebook 实例中运行“import CV2”。但在推理期间未能在端点中运行它。我尝试了以下方法，使用“env”作为链接AWS Sagemaker - Install External Library and Make it Persist

但它仍然无法正常工作。

谁有解决的好方法？谢谢！

我的代码是：

env = {
'SAGEMAKER_REQUIREMENTS': 'requirements.txt', # path relative to `source_dir` below.
}
image_embed_model = MXNetModel(model_data=model_data,
                         entry_point='sagemaker_entrypoint.py',
                        role=role,
                        source_dir = 'src',
                        env = env,
                        py_version='py3',
                        framework_version='1.6.0')

transformer = image_embed_model.transformer(instance_count=1, # Please pay attention here!!!
                                    instance_type='ml.m4.xlarge',
                                    output_path=output_path,
                                    assemble_with = 'Line', 
                                    accept = 'text/csv'
                                   )
transformer.transform(batch_input,
                  content_type='text/csv', 
                  split_type='Line',
                  input_filter='$[0:]',
                  join_source='Input',
                  wait=False)

【问题讨论】：

标签： python-import batch-processing endpoint cv2 amazon-sagemaker

【解决方案1】：

您可以关注https://github.com/aws/sagemaker-python-sdk/blob/master/doc/using_mxnet.rst#use-third-party-libraries 将第三方库导入您的批量转换实例。在打包模型数据之前，请确保requirement.txt文件保存在正确的目录下。

【讨论】：