adding rocky8 and rocky9 cuda images

WIPACrepo · Oct 16, 2023 · 4178957 · 4178957
1 parent 248cedf
commit 4178957
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 4 deletions.
diff --git a/.github/workflows/build_base.yml b/.github/workflows/build_base.yml
@@ -164,6 +164,18 @@ jobs:
           tags: >
             icecube/icetray-base:devel-rocky8,
             icecube/icetray-base:devel-rocky8-${{ env.current_date }}
+      -
+        name: Build and push base devel rocky8 CUDA 
+        uses: docker/build-push-action@v3
+        with:
+          push: true
+          file: icetray_base/Dockerfile_base-devel-rocky8
+          platforms: ${{ env.PLATFORMS }}
+          build-args: |
+            BASE_IMAGE=nvidia/cuda:${{env.CUDA_VERSION}}-cudnn${{env.CUDNN_VERSION}}-devel-rockylinux8
+          tags: >
+            icecube/icetray-base:devel-rocky8-cuda${{env.CUDA_VERSION}}-cudnn${{env.CUDNN_VERSION}},
+            icecube/icetray-base:devel-rocky8-cuda${{env.CUDA_VERSION}}-cudnn${{env.CUDNN_VERSION}}-${{ env.current_date }}
       -
         name: Build and push base devel rocky 9
         uses: docker/build-push-action@v3
@@ -174,6 +186,18 @@ jobs:
           tags: >
             icecube/icetray-base:devel-rocky9,
             icecube/icetray-base:devel-rocky9-${{ env.current_date }}
+      -
+        name: Build and push base devel rocky9 CUDA 
+        uses: docker/build-push-action@v3
+        with:
+          push: true
+          file: icetray_base/Dockerfile_base-devel-rocky9
+          platforms: ${{ env.PLATFORMS }}
+          build-args: |
+            BASE_IMAGE=nvidia/cuda:${{env.CUDA_VERSION}}-cudnn${{env.CUDNN_VERSION}}-devel-rockylinux9
+          tags: >
+            icecube/icetray-base:devel-rocky9-cuda${{env.CUDA_VERSION}}-cudnn${{env.CUDNN_VERSION}},
+            icecube/icetray-base:devel-rocky9-cuda${{env.CUDA_VERSION}}-cudnn${{env.CUDNN_VERSION}}-${{ env.current_date }}
       # -
       #   name: Build and push base devel debian sid
       #   uses: docker/build-push-action@v3

diff --git a/icetray_base/Dockerfile_base-devel-rocky8 b/icetray_base/Dockerfile_base-devel-rocky8
@@ -1,8 +1,8 @@
 #### Create a "developer" base for Rocky 8
 # Why Rocky. NVIDIA uses Rocky 8 as the base image
 # This is mostly meant as a image for grid computing
-
-FROM rockylinux:8 as base-devel
+ARG BASE_IMAGE=rockylinux:8
+FROM ${BASE_IMAGE} as base-devel
 
 MAINTAINER IceCube <[email protected]>
 

diff --git a/icetray_base/Dockerfile_base-devel-rocky9 b/icetray_base/Dockerfile_base-devel-rocky9
@@ -1,8 +1,8 @@
 #### Create a "developer" base for Rocky 9
 # Why Rocky. NVIDIA uses Rocky 8 as the base image
 # This is mostly meant as a image for grid computing
-
-FROM rockylinux:9 as base-devel
+ARG BASE_IMAGE=rockylinux:9
+FROM ${BASE_IMAGE} as base-devel
 
 MAINTAINER IceCube <[email protected]>