Add rocker build for cuda

Author: almahmoud

.github/workflows/rocker-builds.yaml

@@ -267,3 +267,91 @@ jobs:
         push: true
         tags: ${{ steps.defs.outputs.rockerintermediateprefix }}-shiny:${{ steps.defs.outputs.rver }}-${{ matrix.build.arch }}
         platforms: linux/${{ matrix.build.arch }}
+
+  tag-versioned-cuda:
+    strategy:
+      matrix:
+        build:
+          - { arch: 'amd64', runner: 'ubuntu-latest', cudatag: 'latest' }
+          - { arch: 'arm64', runner: 'ubuntu-latest-arm64', cudatag: 'arm64' }
+    runs-on: ${{ matrix.build.runner }}
+    steps:
+    - uses: actions/checkout@v4
+
+    - name: Free root space
+      uses: almahmoud/free-root-space@main
+      with:
+        verbose: true
+
+    - name: Set up Docker Buildx
+      uses: docker/setup-buildx-action@v3
+
+    - name: Login to GHCR
+      uses: docker/login-action@v3
+      with:
+        registry: ghcr.io
+        username: ${{ github.actor }}
+        password: ${{ secrets.GITHUB_TOKEN }}
+
+    - name: Pull rocker/cuda image
+      run: |
+        docker pull rocker/cuda:${{ matrix.build.cudatag }}
+
+    - name: Extract R version and CUDA version
+      id: versions
+      run: |
+        # Start container in detached mode to avoid repulling
+        CONTAINER_ID=$(docker run -d rocker/cuda:${{ matrix.build.cudatag }} sleep infinity)
+        echo "Started container: $CONTAINER_ID"
+        
+        # Extract R version by running Rscript in the container
+        R_VER=$(docker exec "$CONTAINER_ID" Rscript -e 'cat(as.character(getRversion()))' 2>/dev/null || echo "unknown")
+        echo "r_version=$R_VER" >> $GITHUB_OUTPUT
+        echo "Detected R version: $R_VER"
+        
+        # Extract CUDA version by running pytorch in the container
+        CUDA_VER=$(docker exec "$CONTAINER_ID" python3 -c "import torch; print(torch.version.cuda)" 2>/dev/null || echo "unknown")
+        echo "cuda_version=$CUDA_VER" >> $GITHUB_OUTPUT
+        echo "Detected CUDA version: $CUDA_VER"
+        
+        # Stop and remove the container
+        docker stop "$CONTAINER_ID"
+        docker rm "$CONTAINER_ID"
+        
+        # Get lowercase repository owner
+        OWNER_LOWER=$(echo "${{ github.repository_owner }}" | awk '{print tolower($0)}')
+        echo "owner=$OWNER_LOWER" >> $GITHUB_OUTPUT
+
+    - name: Tag and push to GHCR with versioned tags
+      run: |
+        SOURCE_IMAGE="rocker/cuda:${{ matrix.build.cudatag }}"
+        BASE_TAG="ghcr.io/${{ steps.versions.outputs.owner }}/rocker-cuda"
+        ARCH="${{ matrix.build.arch }}"
+        R_VER="${{ steps.versions.outputs.r_version }}"
+        CUDA_VER="${{ steps.versions.outputs.cuda_version }}"
+        
+        # Tag with architecture
+        docker tag "$SOURCE_IMAGE" "${BASE_TAG}:${ARCH}"
+        docker push "${BASE_TAG}:${ARCH}"
+        echo "Pushed: ${BASE_TAG}:${ARCH}"
+        
+        # Tag with R version
+        if [ ! -z "$R_VER" ] && [ "$R_VER" != "unknown" ]; then
+          docker tag "$SOURCE_IMAGE" "${BASE_TAG}:r-${R_VER}-${ARCH}"
+          docker push "${BASE_TAG}:r-${R_VER}-${ARCH}"
+          echo "Pushed: ${BASE_TAG}:r-${R_VER}-${ARCH}"
+        fi
+        
+        # Tag with CUDA version
+        if [ ! -z "$CUDA_VER" ] && [ "$CUDA_VER" != "unknown" ]; then
+          docker tag "$SOURCE_IMAGE" "${BASE_TAG}:cuda-${CUDA_VER}-${ARCH}"
+          docker push "${BASE_TAG}:cuda-${CUDA_VER}-${ARCH}"
+          echo "Pushed: ${BASE_TAG}:cuda-${CUDA_VER}-${ARCH}"
+        fi
+        
+        # Tag with both R and CUDA versions
+        if [ ! -z "$R_VER" ] && [ "$R_VER" != "unknown" ] && [ ! -z "$CUDA_VER" ] && [ "$CUDA_VER" != "unknown" ]; then
+          docker tag "$SOURCE_IMAGE" "${BASE_TAG}:r-${R_VER}-cuda-${CUDA_VER}-${ARCH}"
+          docker push "${BASE_TAG}:r-${R_VER}-cuda-${CUDA_VER}-${ARCH}"
+          echo "Pushed: ${BASE_TAG}:r-${R_VER}-cuda-${CUDA_VER}-${ARCH}"
+        fi