Add PATH for tensorboard

GoogleCloudPlatform · Jul 12, 2024 · 2adf4e2 · 2adf4e2
1 parent 1d87e76
commit 2adf4e2
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/notebooks/image_models/labs/stable_diffusion_dreambooth.ipynb b/notebooks/image_models/labs/stable_diffusion_dreambooth.ipynb
@@ -62,7 +62,11 @@
     "import warnings\n",
     "\n",
     "os.environ[\"TF_CPP_MIN_LOG_LEVEL\"] = \"2\"\n",
-    "warnings.filterwarnings(\"ignore\")"
+    "warnings.filterwarnings(\"ignore\")\n",
+    "\n",
+    "# Set `PATH` to include the directory containing tensorboard\n",
+    "PATH = %env PATH\n",
+    "%env PATH=/home/jupyter/.local/bin:{PATH}"
    ]
   },
   {

diff --git a/notebooks/image_models/solutions/stable_diffusion_dreambooth.ipynb b/notebooks/image_models/solutions/stable_diffusion_dreambooth.ipynb
@@ -62,7 +62,11 @@
     "import warnings\n",
     "\n",
     "os.environ[\"TF_CPP_MIN_LOG_LEVEL\"] = \"2\"\n",
-    "warnings.filterwarnings(\"ignore\")"
+    "warnings.filterwarnings(\"ignore\")\n",
+    "\n",
+    "# Set `PATH` to include the directory containing tensorboard\n",
+    "PATH = %env PATH\n",
+    "%env PATH=/home/jupyter/.local/bin:{PATH}"
    ]
   },
   {