File size: 2,188 Bytes
21f17cc
2734e3f
 
 
 
 
 
 
 
 
 
990bec6
2734e3f
f950a88
d35278a
f950a88
 
48612f8
f950a88
c5b0af1
e0011fd
c43c5c8
 
 
c5b0af1
e0011fd
6ef96f5
48612f8
f950a88
c5b0af1
f950a88
6ef96f5
48612f8
6ef96f5
e0011fd
6ef96f5
 
2734e3f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
c43c5c8
2734e3f
990bec6
 
f950a88
 
 
c43c5c8
f950a88
6ef96f5
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
name: ci-cd-base

on:
  push:
    branches:
      - "main-base"
      - "dev-base"

jobs:
  build-base:
    if: github.repository_owner == 'OpenAccess-AI-Collective'
    # this job needs to be run on self-hosted GPU runners...
    runs-on: self-hosted
    strategy:
      fail-fast: false
      matrix:
        include:
          - cuda: "118"
            cuda_version: 11.8.0
            python_version: "3.9"
            pytorch: 2.0.0
            axolotl_extras:
          - cuda: "118"
            cuda_version: 11.8.0
            python_version: "3.10"
            pytorch: 2.0.0
            axolotl_extras:
          - cuda: "117"
            cuda_version: 11.7.0
            python_version: "3.9"
            pytorch: 1.13.1
            axolotl_extras:
          - cuda: "118"
            cuda_version: 11.8.0
            python_version: "3.9"
            pytorch: 2.0.0
            axolotl_extras: gptq
    steps:
      - name: Checkout
        uses: actions/checkout@v3
      - name: Docker metadata
        id: metadata
        uses: docker/metadata-action@v3
        with:
          images: winglian/axolotl-base
      - name: Login to Docker Hub
        uses: docker/login-action@v2
        with:
          username: ${{ secrets.DOCKERHUB_USERNAME }}
          password: ${{ secrets.DOCKERHUB_TOKEN }}
      - name: Set up Docker Buildx
        uses: docker/setup-buildx-action@v2
      - name: Build
        uses: docker/build-push-action@v4
        with:
          context: .
          file: ./docker/Dockerfile-base
          push: ${{ github.event_name != 'pull_request' }}
          tags: ${{ steps.metadata.outputs.tags }}-py${{ matrix.python_version }}-cu${{ matrix.cuda }}-${{ matrix.pytorch }}${{ matrix.axolotl_extras != '' && '-' || '' }}${{ matrix.axolotl_extras }}
          labels: ${{ steps.metadata.outputs.labels }}
          cache-from: type=gha
          cache-to: type=gha,mode=max
          build-args: |
            CUDA_VERSION=${{ matrix.cuda_version }}
            CUDA=${{ matrix.cuda }}
            PYTHON_VERSION=${{ matrix.python_version }}
            PYTORCH_VERSION=${{ matrix.pytorch }}
            AXOLOTL_EXTRAS=${{ matrix.axolotl_extras }}