update container_uri to match update

Author: glasnt

people-and-planet-ai/weather-forecasting/notebooks/3-training.ipynb

@@ -1381,7 +1381,7 @@
     "    display_name=\"weather-forecasting\",\n",
     "    python_package_gcs_uri=f\"gs://{bucket}/weather/weather-model-1.0.0.tar.gz\",\n",
     "    python_module_name=\"weather.trainer\",\n",
-    "    container_uri=\"us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-4.py310:latest\",\n",
+    "    container_uri=\"us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-8.py310:latest\",\n",
     ")\n",
     "job.run(\n",
     "    machine_type=\"n1-highmem-8\",\n",