Update multiple_frameworks_lab.ipynb

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