Skip to content

Torch dependency in sagameker-core to be made optional #5457

@sateeshmannar

Description

@sateeshmannar

PySDK Version

  • PySDK V2 (2.x)
  • PySDK V3 (3.x)

Describe the bug
Can torch dependency be made optional? Torch feature is NOT used by all users and adding it as a dependency brings in a lot of additional transitive dependencies resulting in any custom image built w/sagemaker-core to be bloated. Can this be made optional?

To reproduce
pip install sagemaker-core==2.3.0

Expected behavior
Sagemaker-core install should not install torch by default

Screenshots or logs
NA

System information
A description of your system. Please provide:

  • SageMaker Python SDK version: 3.3.0
  • Framework name (eg. PyTorch) or algorithm (eg. KMeans): Torch
  • Framework version: NA
  • Python version: 3.1.13
  • CPU or GPU: CPU
  • Custom Docker image (Y/N): Y

Additional context
We are trying to keep our custom images as lean as possible and avoid includng packages that are not used in our env.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions