Fixed torch version

isaac-sim · Jun 1, 2024 · 7ceccc0 · 7ceccc0
1 parent 5aacc16
commit 7ceccc0
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/docs/source/setup/installation/pip_installation.rst b/docs/source/setup/installation/pip_installation.rst
@@ -36,21 +36,21 @@ Installing Isaac Sim
             isaaclab\Scripts\activate
 
 
--  Next, install a CUDA-enabled PyTorch 2.3.0 build based on the CUDA version available on your system.
+-  Next, install a CUDA-enabled PyTorch 2.2.2 build based on the CUDA version available on your system.
 
    .. tabs::
 
       .. tab:: CUDA 11
 
          .. code-block:: bash
 
-            pip install torch --index-url https://download.pytorch.org/whl/cu118
+            pip install torch==2.2.2 --index-url https://download.pytorch.org/whl/cu118
 
       .. tab:: CUDA 12
 
          .. code-block:: bash
 
-            pip install torch
+            pip install torch==2.2.2 --index-url https://download.pytorch.org/whl/cu121
 
 
 -  Then, install the Isaac Sim packages necessary for running Isaac Lab:

diff --git a/source/extensions/omni.isaac.lab/setup.py b/source/extensions/omni.isaac.lab/setup.py
@@ -19,7 +19,7 @@
 INSTALL_REQUIRES = [
     # generic
     "numpy",
-    "torch==2.3.0",
+    "torch>=2.2.2",
     "prettytable==3.3.0",
     "tensordict",
     "toml",

diff --git a/source/extensions/omni.isaac.lab_tasks/setup.py b/source/extensions/omni.isaac.lab_tasks/setup.py
@@ -21,7 +21,7 @@
 INSTALL_REQUIRES = [
     # generic
     "numpy",
-    "torch==2.3.0",
+    "torch>=2.2.2",
     "torchvision>=0.14.1",  # ensure compatibility with torch 1.13.1
     # 5.26.0 introduced a breaking change, so we restricted it for now.
     # See issue https://github.com/tensorflow/tensorboard/issues/6808 for details.