Update multiple_frameworks_kubeflow.ipynb

GoogleCloudPlatform · Feb 6, 2024 · 69f7334 · 69f7334
1 parent 786de21
commit 69f7334
Showing 1 changed file with 4 additions and 4 deletions.
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",