[testing] docker workflows

.github/workflows/build-container.yml

@@ -0,0 +1,78 @@
+name: Build Docker Container
+on:
+  pull_request:
+    types: [opened, synchronize, reopened]
+    branches:
+      - main
+      - 'release/**'
+  push:
+    branches:
+      - 'main'
+  release:
+    types: [created, published]
+  schedule:
+    - cron: '0 2 * * *'
+
+# potentially update to change criteria; only run when wheel build is complete
+# use outptus from wheel build to determie if release/dev/nightly
+env:
+  RELEASE: ${{ github.event_name =='release' || (github.base_ref == 'release/**' && github.event_name == 'pull_request')}}
+  DEV: ${{ github.base_ref == 'main' && github.event_name == 'pull_request'}}
+  NAME: ${{ github.event.number }} 
+
+permissions:
+  contents: read
+  packages: write
+
+jobs:
+  build-container:
+    name: Build sparseml container
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v3
+        with:
+          fetch-depth: 1
+      - name: Set up Docker Buildx
+        id: buildx
+        uses: docker/setup-buildx-action@v2
+        with:
+          buildkitd-flags: --debug
+      - name: Get current date
+        id: date
+        run: echo "::set-output name=date::$(date +'%Y%m%d')"
+      - name: Get the current version
+        if: ${{ env.RELEASE == 'true' }}
+        id: version
+        run: echo "##[set-output name=version;]$(echo ${GITHUB_REF_NAME#*/})"
+      - name: Login to Github Packages
+        uses: docker/login-action@v2
+        with:
+          registry: ghcr.io
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+      - name: Build Dev Docker Container
+        if: ${{ env.DEV == 'true' }}
+        uses: docker/build-push-action@v4
+        with: 
+          context: ./docker/containers/docker_dev
+          build-args: |
+            BRANCH=${{github.head_ref}}
+          push: true
+          tags: ghcr.io/neuralmagic/test-sparseml-dev:${{ env.NAME }}
+      - name: Build RELEASE Docker Container
+        if: ${{ env.RELEASE == 'true' }}
+        uses: docker/build-push-action@v4
+        with: 
+          context: ./docker/containers/docker_release
+          build-args: |
+            VERSION=${{ steps.version.outputs.version }}
+          push: true
+          tags: ghcr.io/neuralmagic/test-sparseml:latest, ghcr.io/neuralmagic/test-sparseml:${{ steps.version.outputs.version }}
+      - name: Build Nightly Docker Container
+        if: ${{ env.DEV == 'false' && env.RELEASE == 'false'}}
+        uses: docker/build-push-action@v4
+        with:
+          context: ./docker/containers/docker_nightly
+          push: true
+          tags: ghcr.io/neuralmagic/test-sparseml-nightly:latest, ghcr.io/neuralmagic/test-sparseml-nightly:${{ steps.date.outputs.date }}

docker/Dockerfile

@@ -93,7 +93,7 @@ RUN \
     fi;
 
 FROM cuda_builder AS container_branch_all
-ARG VENV
+ARG VENV 
 ENV PATH="${VENV}/bin:$PATH"
 ENV PIP_DEFAULT_TIMEOUT=200
 ARG VERSION

docker/containers/docker_dev/Dockerfile

@@ -0,0 +1,15 @@
+ARG SOURCE=ghcr.io/neuralmagic/cuda-python3.8
+ARG BRANCH
+
+FROM $SOURCE
+
+ARG BRANCH
+
+RUN python3.8 -m pip install --upgrade pip \
+    && python3.8 -m pip install --upgrade setuptools \
+    && git clone https://github.com/neuralmagic/sparseml.git --depth 1 --single-branch -b ${BRANCH} \
+    && python3.8 -m pip install -e "./sparseml[dev]"
+
+HEALTHCHECK CMD python3.8 -c 'import sparseml'
+RUN python3.8 -m pip list | grep sparseml
+CMD bash

docker/containers/docker_nightly/Dockerfile

@@ -0,0 +1,11 @@
+ARG SOURCE=ghcr.io/neuralmagic/cuda-python3.8
+
+FROM $SOURCE
+
+RUN python3.8 -m pip install --upgrade pip \
+    && python3.8 -m pip install --upgrade setuptools \
+    && python3.8 -m pip install --no-cache-dir "sparseml-nightly[onnxruntime,torchvision,transformers,yolov5,ultralytics]" 
+
+HEALTHCHECK CMD python3.8 -c 'import sparseml'
+RUN python3.8 -m pip list | grep sparseml
+CMD bash

docker/containers/docker_release/Dockerfile

@@ -0,0 +1,14 @@
+ARG SOURCE=ghcr.io/neuralmagic/cuda-python3.8
+ARG VERSION
+
+FROM $SOURCE
+
+ARG VERSION
+
+RUN python3.8 -m pip install --upgrade pip \
+    && python3.8 -m pip install --upgrade setuptools \
+    && python3.8 -m pip install --no-cache-dir "sparseml[onnxruntime,torchvision,transformers,yolov5,ultralytics]==$VERSION" 
+
+HEALTHCHECK CMD python3.8 -c 'import sparseml'
+RUN python3.8 -m pip list | grep sparseml
+CMD bash