Google Git
Sign in
android / platform / external / pytorch / aec09eeb3a570bb79c642e8ae540b3981e7d69cd / . / .circleci / docker
tree: a769f2f102f287880324d7f46922f16ed66b197a [path history] [tgz]
  1. android/
  2. centos-rocm/
  3. common/
  4. java/
  5. ubuntu/
  6. ubuntu-cuda/
  7. ubuntu-rocm/
  8. build.sh
  9. build_docker.sh
  10. README.md
  11. requirements-ci.txt
.circleci/docker/README.md

Docker images for Jenkins

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).

Docker builds are now defined with .circleci/cimodel/data/simple/docker_definitions.py

Contents

  • build.sh -- dispatch script to launch all builds
  • common -- scripts used to execute individual Docker build stages
  • ubuntu-cuda -- Dockerfile for Ubuntu image with CUDA support for nvidia-docker

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