Merge pull request #84 from salesforce/pycuda-context

update notebook
salesforce · Jun 16, 2023 · 76e6974 · 76e6974
2 parents 4c6acb2 + d264b53
commit 76e6974
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tutorials/simple-end-to-end-example.ipynb b/tutorials/simple-end-to-end-example.ipynb
@@ -60,7 +60,7 @@
    },
    "outputs": [],
    "source": [
-    "! pip install --quiet \"rl-warp-drive>=2.0\" \"torch==1.10.*\" \"torchvision==0.11.*\" \"torchtext==0.11.*\""
+    "! pip install --quiet \"rl-warp-drive>=2.4\""
    ]
   },
   {

diff --git a/tutorials/tutorial-5-training_with_warp_drive.ipynb b/tutorials/tutorial-5-training_with_warp_drive.ipynb
@@ -84,7 +84,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "! pip install --quiet \"rl-warp-drive>=2.0\" \"torch==1.10.*\" \"torchvision==0.11.*\" \"torchtext==0.11.*\" \"ffmpeg-python\" "
+    "! pip install --quiet \"rl-warp-drive>=2.4\" \"ffmpeg-python\" "
    ]
   },
   {