Open
Description
Checklist
- I've prepended issue tag with type of change: [bug]
- (If applicable) I've attached the script to reproduce the bug
- (If applicable) I've documented below the DLC image/dockerfile this relates to
- (If applicable) I've documented below the tests I've run on the DLC image
- I'm using an existing DLC image listed here: https://docs.aws.amazon.com/deep-learning-containers/latest/devguide/deep-learning-containers-images.html
- I've built my own container based off DLC (and I've attached the code used to build my own image)
Concise Description:
The size of the DLC image is more than the recommended limit on the sagemaker platform
DLC image/dockerfile:
763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.0.0-gpu-py310-cu118-ubuntu20.04-sagemaker
Current behavior:
The endpoint creation fails with reason - Image size 13494688833 is greater than supported size 10737418240
Expected behavior:
The size of the DLC should be less than the size of the recommended container on Sagemaker
Additional context:
I have a pretrained yolov8 model and the code package is total 208 MB including the weights.
Metadata
Metadata
Assignees
Labels
No labels