From 021424fd6744eabec4fe6dbd15c1c950b846adc9 Mon Sep 17 00:00:00 2001
From: Eugene Cheah <PicoCreator@users.noreply.github.com>
Date: Sat, 3 Feb 2024 02:05:16 -0800
Subject: [PATCH] cuda 12-1 build

---
 .github/workflows/docker-build.yml            | 22 +++++++++----------
 .../Dockerfile                                |  2 +-
 2 files changed, 12 insertions(+), 12 deletions(-)
 rename docker/{env-cuda-12-3 => env-cuda-12-1}/Dockerfile (95%)

diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml
index 98996a29..bcb68007 100644
--- a/.github/workflows/docker-build.yml
+++ b/.github/workflows/docker-build.yml
@@ -1,4 +1,4 @@
-name: Docker Env Image (cuda-12-3)
+name: Docker Env Image (cuda-12-1)
 
 on:
   push:
@@ -21,7 +21,7 @@ env:
 
 jobs:
   build_env:
-    name: Docker Env Image (cuda-12-3)
+    name: Docker Env Image (cuda-12-1)
   
     runs-on: ubuntu-latest
     permissions:
@@ -103,20 +103,20 @@ jobs:
       
       # Build and push Docker image with Buildx (don't push on PR)
       # https://github.com/docker/build-push-action
-      - name: Build and push Docker image (env-cuda-12-3)
+      - name: Build and push Docker image (env-cuda-12-1)
         id: build-and-push
         uses: docker/build-push-action@v4
         with:
-          context: "{{defaultContext}}:docker/env-cuda-12-3"
+          context: "{{defaultContext}}:docker/env-cuda-12-1"
           push: ${{ github.event_name != 'pull_request' }} # Don't push on PR
-          tags: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME_LC }}:env-cuda-12-3
+          tags: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME_LC }}:env-cuda-12-1
           # tags: ${{ steps.meta.outputs.tags }}
           labels: ${{ steps.meta.outputs.labels }}
-          cache-from: type=gha,src=docker/env-cuda-12-3
+          cache-from: type=gha,src=docker/env-cuda-12-1
           cache-to: type=gha,mode=max
 
   # build_runner:
-  #   name: Docker Env Image (github-worker-12-3)
+  #   name: Docker Env Image (github-worker-12-1)
   
   #   needs: build_env
   #   runs-on: ubuntu-latest
@@ -199,14 +199,14 @@ jobs:
       
   #     # Build and push Docker image with Buildx (don't push on PR)
   #     # https://github.com/docker/build-push-action
-  #     - name: Build and push Docker image (github-worker-cuda-12-3)
+  #     - name: Build and push Docker image (github-worker-cuda-12-1)
   #       id: build-and-push
   #       uses: docker/build-push-action@v4
   #       with:
-  #         context: "{{defaultContext}}:docker/github-worker-cuda-12-3"
+  #         context: "{{defaultContext}}:docker/github-worker-cuda-12-1"
   #         push: ${{ github.event_name != 'pull_request' }} # Don't push on PR
-  #         tags: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME_LC }}:github-worker-cuda-12-3
+  #         tags: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME_LC }}:github-worker-cuda-12-1
   #         # tags: ${{ steps.meta.outputs.tags }}
   #         labels: ${{ steps.meta.outputs.labels }}
-  #         cache-from: type=gha,src=docker/github-worker-cuda-12-3
+  #         cache-from: type=gha,src=docker/github-worker-cuda-12-1
   #         cache-to: type=gha,mode=max
diff --git a/docker/env-cuda-12-3/Dockerfile b/docker/env-cuda-12-1/Dockerfile
similarity index 95%
rename from docker/env-cuda-12-3/Dockerfile
rename to docker/env-cuda-12-1/Dockerfile
index 9efd337a..d6ac39e5 100644
--- a/docker/env-cuda-12-3/Dockerfile
+++ b/docker/env-cuda-12-1/Dockerfile
@@ -1,6 +1,6 @@
 # We start from the CUDA base image
 # todo: consider other base images?
-FROM nvidia/12.3.1-devel-ubuntu20.04
+FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
 
 # Install ninja, and several common stuff
 RUN apt-get update && \