Fix container image building CI for CUDA 11/12 (#42)

Signed-off-by: Hung-Han (Henry) Chen <chenhungh@gmail.com>

.github/workflows/cuda_image.yaml

@@ -42,7 +42,9 @@ jobs:
           context: .
           file: ./Dockerfile.cuda11
           push: true
-          tags: ${{ steps.meta.outputs.tags }}
+          tags: |
+            ${{ env.REGISTRY }}/${{ github.repository_owner }}/${{ env.CUDA_11_IMAGE_NAME }}:${{ github.sha }}
+            ${{ env.REGISTRY }}/${{ github.repository_owner }}/${{ env.CUDA_11_IMAGE_NAME }}:latest
           labels: ${{ steps.meta.outputs.labels }}
   cuda12_image_to_gcr:
     runs-on: ubuntu-latest
@@ -70,5 +72,7 @@ jobs:
           context: .
           file: ./Dockerfile.cuda12
           push: true
-          tags: ${{ steps.meta.outputs.tags }}
+          tags: |
+            ${{ env.REGISTRY }}/${{ github.repository_owner }}/${{ env.CUDA_12_IMAGE_NAME }}:${{ github.sha }}
+            ${{ env.REGISTRY }}/${{ github.repository_owner }}/${{ env.CUDA_12_IMAGE_NAME }}:latest
           labels: ${{ steps.meta.outputs.labels }}

Dockerfile.cuda12

 # syntax=docker/dockerfile:1
 
-FROM nvidia/cuda:12-base-ubuntu22.04
+FROM nvidia/cuda:12.2.0-base-ubuntu20.04
 RUN apt-get update && apt-get install -y -q python3 python3-pip
 WORKDIR /app
 COPY requirements.txt requirements.txt