aws / aws/sagemaker-python-sdk

SageMaker Local Mode: Error for network sagemaker-local

未关闭
#4,659 2 条评论 1 个 reaction 已指派 1 人 已被 @mufaddal-rohawala 认领 在 GitHub 查看
component: local mode type: bug
主要语言
Python
星标
2.3k
派生
1.3k
平均合并
1 天 22 小时
30 天内合并 PR
35

描述

**Describe the bug**
Running the ScriptProcessor in local mode on either a desktop or notebook instance results in a RuntimeError when it reaches the 'docker compose up' command.

**To reproduce**
First run commands:
docker network rm sagemaker-local
docker network create sagemaker-local

```import boto3

sess = boto3.Session()
sm = sess.client("sagemaker")
s3_resource = boto3.resource("s3")
sagemaker_session = LocalSession()
config = {
"local": {
"local_code": True,
}
}
sagemaker_session.config = config
sagemaker_session.s3_resource = s3_resource
sagemaker_session.s3_client = sm

network_config = NetworkConfig(
enable_network_isolation=False,
security_group_ids=security_group_ids, # security group ids
subnets=subnets, # subnets
encrypt_inter_container_traffic=True,
)

script_processor = ScriptProcessor(
command=["bash", "process.sh"],
image_uri=image, # custom Docker image
role=role,
instance_count=1,
instance_type="local",
sagemaker_session=sagemaker_session,
network_config=network_config,
).run(...)
```

Just running 'docker rm sagemaker-local' and then the code in local mode is a workaround for the issue as of now.

**Expected behavior**
For the script processing job to run and output properly.

**Screenshots or logs**
![image](https://github.com/aws/sagemaker-python-sdk/assets/169085464/815afe9b-0b03-45ba-aa11-460eecca7eca)
![image](https://github.com/aws/sagemaker-python-sdk/assets/169085464/74dd6432-e899-4de7-9c75-f234de14d75e)

**System information**
A description of your system. Please provide:
- **SageMaker Python SDK version**:2.217.0
- **Python version**:3.11.4
- **Custom Docker image**:Y

**Additional context**
Add any other context about the problem here.

贡献指南

打开贡献指南

评估

这个 Issue 还没有评估数据。

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。