forked from pytorch/builder
42 lines (39 loc) · 1.04 KB
/
build-conda-images.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
name: Build conda docker images
on:
push:
branches:
main
paths:
- conda/Dockerfile
- 'common/*'
- .github/workflows/build-conda-images.yml
pull_request:
paths:
- conda/Dockerfile
- 'common/*'
- .github/workflows/build-conda-images.yml
env:
DOCKER_REGISTRY: "docker.io"
DOCKER_BUILDKIT: 1
DOCKER_ID: ${{ secrets.DOCKER_ID }}
DOCKER_TOKEN: ${{ secrets.DOCKER_TOKEN }}
WITH_PUSH: ${{ github.event_name == 'push' && github.ref == 'refs/heads/main' }}
jobs:
build-docker:
runs-on: ubuntu-22.04
strategy:
matrix:
cuda_version: ["11.6", "11.7", "11.8", "cpu"]
env:
CUDA_VERSION: ${{ matrix.cuda_version }}
steps:
- name: Checkout PyTorch builder
uses: actions/checkout@v3
- name: Authenticate if WITH_PUSH
run: |
if [[ "${WITH_PUSH}" == true ]]; then
echo "${DOCKER_TOKEN}" | docker login -u "${DOCKER_ID}" --password-stdin
fi
- name: Build Docker Image
run: |
conda/build_docker.sh