Merge pull request #1 from nidhinpd-YML/gsp-634-bugfix-package

aiplatform package installation, kernel restart code added

Author: dev-nid

self-paced-labs/learning-tensorflow/convolutions-with-complex-images/CLS_Vertex_AI_CNN_horse_or_human.ipynb

@@ -106,6 +106,49 @@
     "## Setting up the environment"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Installation"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "pip install google-cloud-aiplatform"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {
+    "tags": []
+   },
+   "source": [
+    "### Restart the kernel\n",
+    "\n",
+    "Once you've installed everything, you need to restart the notebook kernel so it can find the packages."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import os\n",
+    "\n",
+    "if not os.getenv(\"IS_TESTING\"):\n",
+    "    # Automatically restart kernel after installs\n",
+    "    import IPython\n",
+    "\n",
+    "    app = IPython.Application.instance()\n",
+    "    app.kernel.do_shutdown(True)"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {
@@ -1372,12 +1415,12 @@
   },
   "environment": {
    "kernel": "python3",
-   "name": "tf2-gpu.2-8.m99",
+   "name": "tf2-gpu.2-8.m102",
    "type": "gcloud",
-   "uri": "gcr.io/deeplearning-platform-release/tf2-gpu.2-8:m99"
+   "uri": "gcr.io/deeplearning-platform-release/tf2-gpu.2-8:m102"
   },
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },