Merge pull request #50 from runpod-workers/tagged-releases

feat: auto build cuda version

Author: alpayariyak

.github/workflows/docker-build-release.yml

@@ -19,6 +19,10 @@ jobs:
     # DO is a custom runner deployed on DigitalOcean, only available for workflows under the runpod-workers organization.
     # If you would like to use this workflow, you can replace DO with ubuntu-latest or any other runner.
 
+    strategy:
+      matrix:
+        cuda_version: [11.8.0, 12.1.0]
+
     steps:
       - name: Set up QEMU
         uses: docker/setup-qemu-action@v2
@@ -37,4 +41,5 @@ jobs:
         uses: docker/build-push-action@v4
         with:
           push: true
-          tags: ${{ vars.DOCKERHUB_REPO }}/${{ vars.DOCKERHUB_IMG }}:${{ (github.event_name == 'release' && github.event.release.tag_name) || (github.event_name == 'workflow_dispatch' && github.event.inputs.image_tag) || 'dev' }}
+          tags: ${{ vars.DOCKERHUB_REPO }}/${{ vars.DOCKERHUB_IMG }}:${{ (github.event_name == 'release' && github.event.release.tag_name) || (github.event_name == 'workflow_dispatch' && github.event.inputs.image_tag) || 'dev' }}-cuda${{ matrix.cuda_version }}
+          build-args: WORKER_CUDA_VERSION=${{ matrix.cuda_version }}