fix tutorial

open-mmlab · May 15, 2023 · 50b1886 · 50b1886
1 parent 6c71805
commit 50b1886
Showing 1 changed file with 26 additions and 9 deletions.
diff --git a/demo/mmaction2_tutorial.ipynb b/demo/mmaction2_tutorial.ipynb
@@ -12,6 +12,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "VcjSRFELVbNk"
@@ -28,6 +29,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "7LqHGkGEVqpm"
@@ -71,6 +73,16 @@
         "!gcc --version"
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "# install dependencies: (if your colab has CUDA 11.8)\n",
+        "%pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118"
+      ]
+    },
     {
       "cell_type": "code",
       "execution_count": 2,
@@ -329,14 +341,10 @@
         }
       ],
       "source": [
-        "# install dependencies: (use cu111 because colab has CUDA 11.1)\n",
-        "!pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html\n",
-        "\n",
-        "# Install mmcv\n",
-        "!pip install 'mmcv>=2.0.0' -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html\n",
-        "\n",
-        "# Install mmengine\n",
-        "!pip install mmengine\n",
+        "# install MMEngine, MMCV and MMDetection using MIM\n",
+        "%pip install -U openmim\n",
+        "!mim install mmengine\n",
+        "!mim install \"mmcv>=2.0.0\"\n",
         "\n",
         "# Install mmaction2\n",
         "!rm -rf mmaction2\n",
@@ -392,6 +400,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "pXf7oV5DWdab"
@@ -524,6 +533,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "QuZG8kZ2fJ5d"
@@ -538,6 +548,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "neEFyxChfgiJ"
@@ -734,6 +745,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "0bq0mxmEi29H"
@@ -743,6 +755,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "Ht_DGJA9jQar"
@@ -766,6 +779,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "tc8YhFFGjp3e"
@@ -1026,6 +1040,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "tES-qnZ3k38Z"
@@ -1715,6 +1730,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "zdSd7oTLlxIf"
@@ -1734,6 +1750,7 @@
       ]
     },
     {
+      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "ryVoSfZVmogw"
@@ -1800,7 +1817,7 @@
       "name": "python",
       "nbconvert_exporter": "python",
       "pygments_lexer": "ipython3",
-      "version": "3.7.16"
+      "version": "3.7.12"
     },
     "vscode": {
       "interpreter": {