Google Git
Sign in
android / platform / external / pytorch / 4c70ab26ef0a596d779ec38e9bc9c1bfc4373f93 / . / .ci / docker
tree: 83bcff6ed2cd2c5a39a8c7151cc0ec1614959c12 [path history] [tgz]
  1. android/
  2. centos-rocm/
  3. ci_commit_pins/
  4. common/
  5. java/
  6. linter/
  7. linter-cuda/
  8. ubuntu/
  9. ubuntu-cuda/
  10. ubuntu-rocm/
  11. ubuntu-xpu/
  12. build.sh
  13. README.md
  14. requirements-ci.txt
  15. requirements-docs.txt
  16. triton_version.txt
.ci/docker/README.md

Docker images for GitHub CI

This directory contains everything needed to build the Docker images that are used in our CI.

The Dockerfiles located in subdirectories are parameterized to conditionally run build stages depending on build arguments passed to docker build. This lets us use only a few Dockerfiles for many images. The different configurations are identified by a freeform string that we call a build environment. This string is persisted in each image as the BUILD_ENVIRONMENT environment variable.

See build.sh for valid build environments (it's the giant switch).

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
  • ubuntu-cuda -- Dockerfile for Ubuntu image with CUDA support for nvidia-docker
  • ubuntu-rocm -- Dockerfile for Ubuntu image with ROCm support
  • ubuntu-xpu -- Dockerfile for Ubuntu image with XPU support

Usage

# Build a specific image
./build.sh pytorch-linux-bionic-py3.8-gcc9 -t myimage:latest

# Set flags (see build.sh) and build image
sudo bash -c 'PROTOBUF=1 ./build.sh pytorch-linux-bionic-py3.8-gcc9 -t myimage:latest