To build all docker images you can use the convenience script:
# Build without pushing
manywheel/build_all_docker.sh
# Build with pushing
WITH_PUSH=true manywheel/build_all_docker.sh
To build a specific docker image use:
# GPU_ARCH_TYPE can be ["cuda", "rocm", "cpu"]
# GPU_ARCH_VERSION is GPU_ARCH_TYPE dependent, see manywheel/build_all_docker.sh for examples
GPU_ARCH_TYPE=cuda GPU_ARCH_VERSION=11.1 manywheel/build_docker.sh
# Build with pushing
WITH_PUSH=true GPU_ARCH_TYPE=cuda GPU_ARCH_VERSION=11.1 manywheel/build_docker.sh
DISCLAIMER for WITH_PUSH:
If you'd like to use WITH_PUSH, you must set it to exactly true
, not 1
nor ON
nor even TRUE
, as our scripts
check for exact string equality to enable push functionality. The reason for this rigidity is due to the how we
calculate WITH_PUSH in our GHA workflow YAMLs. Currently, we usually enable push based on the workflow trigger, which
when we query with an expression like ${{ github.event_name == 'push' }}
returns either true
or false
. Thus, we
adapt our scripts to fit with this model.