Docker images for TorchTitan CI

This directory contains everything needed to build the Docker images that are used in TorchTitan CI. The content of this directory are copied from PyTorch CI https://github.com/pytorch/pytorch/tree/main/.ci/docker. It also uses the same directory structure as PyTorch.

Contents

• build.sh -- dispatch script to launch all builds
• common -- scripts used to execute individual Docker build stages
• ubuntu -- Dockerfile for Ubuntu image for CPU build and test jobs

Usage

# Generic usage
./build.sh "${IMAGE_NAME}" "${DOCKER_BUILD_PARAMETERS}"

# Build a specific image
./build.sh torchtitan-ubuntu-22.04-clang12 -t myimage:latest