Merge pull request #206 from GoogleCloudPlatform/ajayhemnani-patch-5

Update multiple_frameworks_kubeflow.ipynb
GoogleCloudPlatform · Feb 6, 2024 · 6a3fcd9 · 6a3fcd9
2 parents 786de21 + deadbbe
commit 6a3fcd9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git 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",

diff --git 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",