From 69f7334196c2d1ea21c222002a1debf2fe9b4bf7 Mon Sep 17 00:00:00 2001 From: Ajay C Hemnani Date: Mon, 5 Feb 2024 16:03:14 -0800 Subject: [PATCH 1/2] Update multiple_frameworks_kubeflow.ipynb --- .../kubeflow/solutions/multiple_frameworks_kubeflow.ipynb | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/continuous_training/kubeflow/solutions/multiple_frameworks_kubeflow.ipynb b/continuous_training/kubeflow/solutions/multiple_frameworks_kubeflow.ipynb index ab32e9f9..f0f44138 100644 --- a/continuous_training/kubeflow/solutions/multiple_frameworks_kubeflow.ipynb +++ b/continuous_training/kubeflow/solutions/multiple_frameworks_kubeflow.ipynb @@ -205,7 +205,7 @@ "source": [ "%%writefile ./scikit_trainer_image/Dockerfile\n", "\n", - "FROM gcr.io/deeplearning-platform-release/base-cpu\n", + "FROM gcr.io/deeplearning-platform-release/base-cpu.py37\n", "RUN pip install -U fire scikit-learn==0.23.2 pandas==1.1.1\n", "WORKDIR /app\n", "COPY train.py .\n", @@ -408,7 +408,7 @@ "source": [ "%%writefile ./tensorflow_trainer_image/Dockerfile\n", "\n", - "FROM gcr.io/deeplearning-platform-release/base-cpu\n", + "FROM gcr.io/deeplearning-platform-release/base-cpu.py37\n", "RUN pip install -U fire tensorflow==2.1.1\n", "WORKDIR /app\n", "COPY train.py .\n", @@ -621,7 +621,7 @@ "source": [ "%%writefile ./pytorch_trainer_image/Dockerfile\n", "\n", - "FROM gcr.io/deeplearning-platform-release/base-cpu\n", + "FROM gcr.io/deeplearning-platform-release/base-cpu.py37\n", "RUN pip install -U fire torch==1.6.0 scikit-learn==0.23.2 pandas==1.1.1\n", "WORKDIR /app\n", "COPY train.py .\n", @@ -755,7 +755,7 @@ "source": [ "%%writefile ./xgboost_trainer_image/Dockerfile\n", "\n", - "FROM gcr.io/deeplearning-platform-release/base-cpu\n", + "FROM gcr.io/deeplearning-platform-release/base-cpu.py37\n", "RUN pip install -U fire scikit-learn==0.23.2 pandas==1.1.1 xgboost==1.2.0\n", "WORKDIR /app\n", "COPY train.py .\n", From deadbbe8fc4239fef57f9d9f5fd490270d5f5e2a Mon Sep 17 00:00:00 2001 From: Ajay C Hemnani Date: Mon, 5 Feb 2024 16:24:19 -0800 Subject: [PATCH 2/2] Update multiple_frameworks_lab.ipynb --- .../kubeflow/labs/multiple_frameworks_lab.ipynb | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/continuous_training/kubeflow/labs/multiple_frameworks_lab.ipynb b/continuous_training/kubeflow/labs/multiple_frameworks_lab.ipynb index cbcd7c50..9234e087 100644 --- a/continuous_training/kubeflow/labs/multiple_frameworks_lab.ipynb +++ b/continuous_training/kubeflow/labs/multiple_frameworks_lab.ipynb @@ -213,7 +213,7 @@ "source": [ "%%writefile ./scikit_trainer_image/Dockerfile\n", "\n", - "FROM gcr.io/deeplearning-platform-release/base-cpu\n", + "FROM gcr.io/deeplearning-platform-release/base-cpu.py37\n", "RUN pip install -U fire scikit-learn==0.23.2 pandas==1.1.1\n", "WORKDIR /app\n", "COPY train.py .\n", @@ -416,7 +416,7 @@ "source": [ "%%writefile ./tensorflow_trainer_image/Dockerfile\n", "\n", - "FROM gcr.io/deeplearning-platform-release/base-cpu\n", + "FROM gcr.io/deeplearning-platform-release/base-cpu.py37\n", "RUN pip install -U fire tensorflow==2.1.1\n", "WORKDIR /app\n", "COPY train.py .\n", @@ -629,7 +629,7 @@ "source": [ "%%writefile ./pytorch_trainer_image/Dockerfile\n", "\n", - "FROM gcr.io/deeplearning-platform-release/base-cpu\n", + "FROM gcr.io/deeplearning-platform-release/base-cpu.py37\n", "RUN pip install -U fire torch==1.6.0 scikit-learn==0.23.2 pandas==1.1.1\n", "WORKDIR /app\n", "COPY train.py .\n", @@ -763,7 +763,7 @@ "source": [ "%%writefile ./xgboost_trainer_image/Dockerfile\n", "\n", - "FROM gcr.io/deeplearning-platform-release/base-cpu\n", + "FROM gcr.io/deeplearning-platform-release/base-cpu.py37\n", "RUN pip install -U fire scikit-learn==0.23.2 pandas==1.1.1 xgboost==1.2.0\n", "WORKDIR /app\n", "COPY train.py .\n",