{0xc00044b600 0xc0004cf0c0} Workflow :: Distributed training with Amazon SageMaker / Amazon EKS Workshop


Navigate to distributed-training-workshop > notebooks > part-2-sagemaker You should see the following files:

├── Dockerfile
├── cpu_eks_cluster.sh
├── gpu_eks_cluster.sh
├── code
│   ├── cifar10-multi-gpu-horovod-k8s.py
│   └── model_def.py
└── specs
    ├── claim-fsx-s3.yaml
    ├── eks_tf_training_job.yaml
    ├── fsx_lustre_policy.json
    └── storage-class-fsx-s3-template.yaml
Files/directories Description
Dockerfile Use this build a custom container image for training on Amazon EKS
cpu_eks_cluster.sh, gpu_eks_cluster.sh shell scripts using eksctl CLI tool to launch an Amazon EKS cluster
code Contains the training scrip and other training script dependencies
specs List of spec files required to configure Kubeflow


We’ll need to first setup Amazon EKS, Amazon FSx for Lustre file system and install Kubeflow. This involves multiple steps and we’ll leverage various CLI tools to to help install, configure and interact with EKS. At a high level, we’ll perform the following steps:

  1. Install eksctl CLI and use it to launch an Amazon EKS cluster
  2. Install kubectl CLI to interact with the Amazon EKS cluster
  3. Install kfclt CLI and use it to configure and install Kubeflow
  4. Allow Amazon EKS to access Amazon FSx for Lustre file system that’s linked to an Amazon S3 bucket
  5. Finally, launch a distributed training job