Fixed build issues

aws · Sep 9, 2022 · 38a90b4 · 38a90b4
1 parent 791ce36
commit 38a90b4
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 8 deletions.
diff --git a/...odel_specific_realtime_analysis/cnn_class_activation_maps/cnn_class_activation_maps.ipynb b/...odel_specific_realtime_analysis/cnn_class_activation_maps/cnn_class_activation_maps.ipynb
@@ -326,6 +326,16 @@
     "Before starting the SageMaker training job, we need to install some libraries. We will use `smdebug` library to read, filter and analyze raw tensors that are stored in Amazon S3. We will use `opencv-python` library to plot saliency maps as heatmap."
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "fab25828",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!apt-get update && apt-get install -y python3-opencv"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,

diff --git a/sagemaker-debugger/pytorch_iterative_model_pruning/iterative_model_pruning_alexnet.ipynb b/sagemaker-debugger/pytorch_iterative_model_pruning/iterative_model_pruning_alexnet.ipynb
@@ -377,7 +377,7 @@
     "    source_dir=\"src\",\n",
     "    entry_point=\"train.py\",\n",
     "    framework_version=\"1.12\",\n",
-    "    py_version=\"py3\",\n",
+    "    py_version=\"py38\",\n",
     "    metric_definitions=[\n",
     "        {\"Name\": \"train:loss\", \"Regex\": \"loss:(.*?)\"},\n",
     "        {\"Name\": \"eval:acc\", \"Regex\": \"acc:(.*?)\"},\n",
@@ -785,9 +785,9 @@
  "metadata": {
   "instance_type": "ml.t3.medium",
   "kernelspec": {
-   "display_name": "Python 3 (PyTorch 1.6 Python 3.6 CPU Optimized)",
+   "display_name": "Python 3.8.11 64-bit ('3.8.11')",
    "language": "python",
-   "name": "python3__SAGEMAKER_INTERNAL__arn:aws:sagemaker:us-west-2:236514542706:image/pytorch-1.6-cpu-py36-ubuntu16.04-v1"
+   "name": "python3"
   },
   "language_info": {
    "codemirror_mode": {
@@ -799,7 +799,12 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.6.13"
+   "version": "3.8.11"
+  },
+  "vscode": {
+   "interpreter": {
+    "hash": "7dc0c77d699a6b3ef5666acd9c55af31cd945013d6c45fb4570db128e5a0b8db"
+   }
   }
  },
  "nbformat": 4,

diff --git a/sagemaker-debugger/pytorch_iterative_model_pruning/iterative_model_pruning_resnet.ipynb b/sagemaker-debugger/pytorch_iterative_model_pruning/iterative_model_pruning_resnet.ipynb
@@ -341,7 +341,7 @@
     "    source_dir=\"src\",\n",
     "    entry_point=\"train.py\",\n",
     "    framework_version=\"1.12\",\n",
-    "    py_version=\"py3\",\n",
+    "    py_version=\"py38\",\n",
     "    metric_definitions=[\n",
     "        {\"Name\": \"train:loss\", \"Regex\": \"loss:(.*?)\"},\n",
     "        {\"Name\": \"eval:acc\", \"Regex\": \"acc:(.*?)\"},\n",
@@ -721,9 +721,9 @@
  "metadata": {
   "instance_type": "ml.t3.medium",
   "kernelspec": {
-   "display_name": "Python 3 (PyTorch 1.6 Python 3.6 CPU Optimized)",
+   "display_name": "Python 3.8.11 64-bit ('3.8.11')",
    "language": "python",
-   "name": "python3__SAGEMAKER_INTERNAL__arn:aws:sagemaker:us-west-2:236514542706:image/pytorch-1.6-cpu-py36-ubuntu16.04-v1"
+   "name": "python3"
   },
   "language_info": {
    "codemirror_mode": {
@@ -735,7 +735,12 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.6.13"
+   "version": "3.8.11"
+  },
+  "vscode": {
+   "interpreter": {
+    "hash": "7dc0c77d699a6b3ef5666acd9c55af31cd945013d6c45fb4570db128e5a0b8db"
+   }
   }
  },
  "nbformat": 4,