Skip to content

SageMaker HyperPod Checkpointless Training v1.0.1

Latest

Choose a tag to compare

@haohanchen-aws haohanchen-aws released this 07 Apr 23:21
08c4ff6

Date: Apr 07, 2026

Bug Fixes

  • Fixed incorrect CUDA device binding in fault handling thread: The fault handling thread now correctly sets the CUDA device context using LOCAL_RANK, preventing device mismatch errors during in-process fault recovery.

Availability

  Currently images are only available in:

  eu-north-1
  ap-south-1
  us-east-2
  eu-west-1
  eu-central-1
  sa-east-1
  us-east-1
  eu-west-2
  ap-northeast-1
  us-west-2
  us-west-1
  ap-southeast-1
  ap-southeast-2

  but not available in the following 3 opt-in Regions:

  ap-southeast-3
  ap-southeast-4
  eu-south-2

Container details

  Checkpointless training Docker container for PyTorch v2.6.0 with CUDA v12.9

Pre-installed packages

  PyTorch: v2.6.0
  CUDA: v12.9
  NCCL: v2.27.5
  EFA: v1.43.0
  AWS-OFI-NCCL v1.16.0
  Libfabric version 2.1
  Megatron v0.15.0
  Nemo v2.6.0rc0