cuda, pytorch matrix for base builds
Browse files- .github/workflows/base.yml +14 -1
.github/workflows/base.yml
CHANGED
|
@@ -11,6 +11,15 @@ jobs:
|
|
| 11 |
if: github.repository_owner == 'OpenAccess-AI-Collective'
|
| 12 |
# this job needs to be run on self-hosted GPU runners...
|
| 13 |
runs-on: self-hosted
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
steps:
|
| 15 |
- name: Checkout
|
| 16 |
uses: actions/checkout@v3
|
|
@@ -32,7 +41,11 @@ jobs:
|
|
| 32 |
context: .
|
| 33 |
file: ./docker/Dockerfile-base
|
| 34 |
push: ${{ github.event_name != 'pull_request' }}
|
| 35 |
-
tags: ${{ steps.metadata.outputs.tags }}
|
| 36 |
labels: ${{ steps.metadata.outputs.labels }}
|
| 37 |
cache-from: type=gha
|
| 38 |
cache-to: type=gha,mode=max
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
if: github.repository_owner == 'OpenAccess-AI-Collective'
|
| 12 |
# this job needs to be run on self-hosted GPU runners...
|
| 13 |
runs-on: self-hosted
|
| 14 |
+
strategy:
|
| 15 |
+
matrix:
|
| 16 |
+
include:
|
| 17 |
+
- cuda: cu118
|
| 18 |
+
cuda_version: 11.8.0
|
| 19 |
+
pytorch: 2.0.0
|
| 20 |
+
- cuda: cu117
|
| 21 |
+
cuda_version: 11.7.0
|
| 22 |
+
pytorch: 1.13.1
|
| 23 |
steps:
|
| 24 |
- name: Checkout
|
| 25 |
uses: actions/checkout@v3
|
|
|
|
| 41 |
context: .
|
| 42 |
file: ./docker/Dockerfile-base
|
| 43 |
push: ${{ github.event_name != 'pull_request' }}
|
| 44 |
+
tags: ${{ steps.metadata.outputs.tags }}-${{ matrix.cuda }}-${{ matrix.pytorch }}
|
| 45 |
labels: ${{ steps.metadata.outputs.labels }}
|
| 46 |
cache-from: type=gha
|
| 47 |
cache-to: type=gha,mode=max
|
| 48 |
+
build-args: |
|
| 49 |
+
CUDA_VERSION=${{ matrix.cuda_version }}
|
| 50 |
+
CUDA=${{ matrix.cuda }}
|
| 51 |
+
PYTORCH_VERSION=${{ matrix.pytorch }}
|