Skip to content

Add 'ml.p5.48xlarge' as a supported instance for SM_EFA_NCCL_INSTANCES.  #219

@andjsmi

Description

@andjsmi

Describe the feature you'd like
Add "ml.p5.48xlarge" to the list of SM_EFA_NCCL_INSTANCES. This means SageMaker Training Toolkit sets FI_PROVIDER=efa automatically for use in training jobs to ensure EFA is used.

How would this feature be used? Please describe.
Automatically setting EFA for use in SageMaker Training Jobs that leverage the SageMaker Training Toolkit

Describe alternatives you've considered
Unsure. Currently you would manually set the environment variable.

Additional context
NA

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