From f170037c1e7642c16cd5fa1707e85abd6252d839 Mon Sep 17 00:00:00 2001
From: cdmikechen <cdmikechen@apache.org>
Date: Fri, 4 Aug 2023 12:40:30 +0800
Subject: [PATCH] SUBMARINE-1400. Change jupyter-gpu cuda image to 11.0.3

### What is this PR for?
Change cuda image to 11.0.3

### What type of PR is it?
Hot Fix

### Todos
* [X] - Change cuda image to 11.0.3

### What is the Jira issue?
https://issues.apache.org/jira/browse/SUBMARINE-1400

### How should this be tested?
CI

### Screenshots (if appropriate)

### Questions:
* Do the license files need updating? No
* Are there breaking changes for older versions? No
* Does this need new documentation? No

Author: cdmikechen <cdmikechen@apache.org>

Signed-off-by: cdmikechen <cdmikechen@apache.org>

Closes #1094 from cdmikechen/SUBMARINE-1400 and squashes the following commits:

f9c3b5ec [cdmikechen] remove image cache
02b02be9 [cdmikechen] change to cuda 11.0.3
---
 .github/workflows/deploy_docker_images.yml    | 44 +++++++++++++------
 .../docker-images/jupyter-gpu/Dockerfile      |  2 +-
 2 files changed, 32 insertions(+), 14 deletions(-)

diff --git a/.github/workflows/deploy_docker_images.yml b/.github/workflows/deploy_docker_images.yml
index 849f49163..76d1144af 100644
--- a/.github/workflows/deploy_docker_images.yml
+++ b/.github/workflows/deploy_docker_images.yml
@@ -53,49 +53,67 @@ jobs:
       - name: Build submarine server
         run: ./dev-support/docker-images/submarine/build.sh
       - name: Push submarine-server docker image
-        run: docker push apache/submarine:server-$SUBMARINE_VERSION
+        run: |
+          docker push apache/submarine:server-$SUBMARINE_VERSION
+          docker rmi apache/submarine:server-$SUBMARINE_VERSION
 
       - name: Build submarine agent
         run: ./dev-support/docker-images/agent/build.sh
       - name: Push submarine-agent docker image
-        run: docker push apache/submarine:agent-$SUBMARINE_VERSION
+        run: |
+          docker push apache/submarine:agent-$SUBMARINE_VERSION
+          docker rmi apache/submarine:agent-$SUBMARINE_VERSION
 
       - name: Build submarine database
         run: ./dev-support/docker-images/database/build.sh
       - name: Push submarine-database docker image
-        run: docker push apache/submarine:database-$SUBMARINE_VERSION
+        run: |
+          docker push apache/submarine:database-$SUBMARINE_VERSION
+          docker rmi apache/submarine:database-$SUBMARINE_VERSION
 
       - name: Build submarine jupyter
         run: ./dev-support/docker-images/jupyter/build.sh
       - name: Push submarine-jupyter docker image
-        run: docker push apache/submarine:jupyter-notebook-$SUBMARINE_VERSION
+        run: |
+          docker push apache/submarine:jupyter-notebook-$SUBMARINE_VERSION
+          docker rmi apache/submarine:jupyter-notebook-$SUBMARINE_VERSION
 
       - name: Build submarine jupyter gpu
         run: ./dev-support/docker-images/jupyter-gpu/build.sh
       - name: Push submarine-jupyter-gpu docker image
-        run: docker push apache/submarine:jupyter-notebook-gpu-$SUBMARINE_VERSION
+        run: |
+          docker push apache/submarine:jupyter-notebook-gpu-$SUBMARINE_VERSION
+          docker rmi apache/submarine:jupyter-notebook-gpu-$SUBMARINE_VERSION
 
       - name: Build submarine operator
         run: ./dev-support/docker-images/operator-v3/build.sh
       - name: Push submarine-operator docker image
-        run: docker push apache/submarine:operator-$SUBMARINE_VERSION
+        run: |
+          docker push apache/submarine:operator-$SUBMARINE_VERSION
+          docker rmi apache/submarine:operator-$SUBMARINE_VERSION
 
       - name: Build submarine mlflow
         run: ./dev-support/docker-images/mlflow/build.sh
       - name: Push submarine-mlflow docker image
-        run: docker push apache/submarine:mlflow-$SUBMARINE_VERSION
+        run: |
+          docker push apache/submarine:mlflow-$SUBMARINE_VERSION
+          docker rmi apache/submarine:mlflow-$SUBMARINE_VERSION
 
-      - name: Build submarine serve
-        run: ./dev-support/docker-images/serve/build.sh
-      - name: Push submarine-serve docker image
-        run: docker push apache/submarine:serve-$SUBMARINE_VERSION
+      # - name: Build submarine serve
+      #   run: ./dev-support/docker-images/serve/build.sh
+      # - name: Push submarine-serve docker image
+      #   run: docker push apache/submarine:serve-$SUBMARINE_VERSION
 
       - name: Build submarine quickstart
         run: ./dev-support/examples/quickstart/build.sh
       - name: Push submarine quickstart docker image
-        run: docker push apache/submarine:quickstart-$SUBMARINE_VERSION
+        run: |
+          docker push apache/submarine:quickstart-$SUBMARINE_VERSION
+          docker rmi apache/submarine:quickstart-$SUBMARINE_VERSION
 
       - name: Build submarine cicd
         run: ./dev-support/cicd/build.sh
       - name: Push submarine-cicd docker image
-        run: docker push apache/submarine:cicd-$SUBMARINE_VERSION
+        run: |
+          docker push apache/submarine:cicd-$SUBMARINE_VERSION
+          docker rmi apache/submarine:cicd-$SUBMARINE_VERSION
diff --git a/dev-support/docker-images/jupyter-gpu/Dockerfile b/dev-support/docker-images/jupyter-gpu/Dockerfile
index d304b5a56..e9004c019 100644
--- a/dev-support/docker-images/jupyter-gpu/Dockerfile
+++ b/dev-support/docker-images/jupyter-gpu/Dockerfile
@@ -13,7 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-FROM nvidia/cuda:12.2.0-base-ubuntu20.04
+FROM nvidia/cuda:11.0.3-base-ubuntu20.04
 
 ARG NB_USER="jovyan"
 ARG NB_UID="1000"