From 4178957323457bc822a300cd1ab8058f5925b887 Mon Sep 17 00:00:00 2001
From: Benedikt Riedel <benedikt.riedel@gmail.com>
Date: Mon, 16 Oct 2023 11:32:49 -0500
Subject: [PATCH] adding rocky8 and rocky9 cuda images

---
 .github/workflows/build_base.yml          | 24 +++++++++++++++++++++++
 icetray_base/Dockerfile_base-devel-rocky8 |  4 ++--
 icetray_base/Dockerfile_base-devel-rocky9 |  4 ++--
 3 files changed, 28 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/build_base.yml b/.github/workflows/build_base.yml
index e03f93b..4f33b7a 100644
--- 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
index 25e2394..5233b79 100644
--- 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 <developers@icecube.wisc.edu>
 
diff --git a/icetray_base/Dockerfile_base-devel-rocky9 b/icetray_base/Dockerfile_base-devel-rocky9
index 2be8952..bc96013 100644
--- 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 <developers@icecube.wisc.edu>