added temporary cell to mmpretrain demo

vaatsalya123 · Feb 20, 2024 · dd8dc8c · dd8dc8c
1 parent 20b1a83
commit dd8dc8c
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/examples_and_demos/mmpretrain_to_jax.ipynb b/examples_and_demos/mmpretrain_to_jax.ipynb
@@ -214,6 +214,19 @@
         "After transpiling our model, we can see what's the improvement in runtime efficiency like. For this let's compile the original PyTorch model using `torch.compile`"
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "# ref : https://github.com/pytorch/pytorch/issues/107960\n",
+        "!export LC_ALL=\"en_US.UTF-8\"\n",
+        "!export LD_LIBRARY_PATH=\"/usr/lib64-nvidia\"\n",
+        "!export LIBRARY_PATH=\"/usr/local/cuda/lib64/stubs\"\n",
+        "!ldconfig /usr/lib64-nvidia"
+      ]
+    },
     {
       "cell_type": "code",
       "execution_count": 8,