[Isaac Sim 6.0] Updates torch to 2.9.0+cu128 for x86#4001
[Isaac Sim 6.0] Updates torch to 2.9.0+cu128 for x86#4001kellyguo11 merged 22 commits intoisaac-sim:feature/isaacsim-6-0from
Conversation
Greptile OverviewGreptile SummaryThis PR updates IsaacLab to be compatible with Isaac Sim 6.0, upgrading PyTorch to 2.9.0 with CUDA 12.8 for x86 platforms and CUDA 13.0 for ARM platforms. Key Changes:
Critical Issue Found: Confidence Score: 1/5
Important Files ChangedFile Analysis
Sequence DiagramsequenceDiagram
participant User
participant Installation Scripts
participant Python Environment
participant Isaac Sim 6.0
participant SimulationContext
participant use_stage Function
participant Stage Utils (core)
participant Stage Utils (experimental)
User->>Installation Scripts: Run isaaclab.sh -i
Installation Scripts->>Python Environment: Install torch 2.9.0+cu128 (x86) or cu130 (ARM)
Installation Scripts->>Python Environment: Install Isaac Sim 6.0
Installation Scripts->>Python Environment: Install IsaacLab dependencies
Note over User,Python Environment: Runtime Behavior
User->>SimulationContext: Initialize SimulationContext
SimulationContext->>SimulationContext: _init_stage()
SimulationContext->>use_stage Function: use_stage(get_initial_stage())
alt Version Check (INVERTED LOGIC BUG)
use_stage Function->>use_stage Function: Check isaac_sim_version < 6
Note over use_stage Function: BUG: Logic is inverted!<br/>For Isaac Sim 6.0, this is FALSE
use_stage Function->>Stage Utils (experimental): Import experimental.utils.stage
use_stage Function->>Stage Utils (core): Set stage context
use_stage Function->>Stage Utils (experimental): Set stage context
end
use_stage Function-->>SimulationContext: Context established
SimulationContext->>Isaac Sim 6.0: super()._init_stage()
SimulationContext->>SimulationContext: _set_additional_physx_params()
SimulationContext->>SimulationContext: _load_fabric_interface()
SimulationContext-->>User: Simulation ready
|
There was a problem hiding this comment.
Additional Comments (3)
-
docs/source/setup/installation/isaaclab_pip_installation.rst, line 28 (link)syntax: comment says CUDA 12.8 but should say CUDA 13.0 to match the cu130 version
-
docs/source/setup/installation/isaaclab_pip_installation.rst, line 66 (link)syntax: outdated PyTorch version in comment - should be 2.9.0, not 2.7.0
-
docs/source/setup/installation/isaaclab_pip_installation.rst, line 73 (link)syntax: outdated PyTorch version - should be 2.9.0, not 2.7.0
7 files reviewed, 8 comments
| .. code-block:: none | ||
|
|
||
| pip install torch==2.7.0 torchvision==0.22.0 --index-url https://download.pytorch.org/whl/cu128 | ||
| pip install torch==2.9.0 torchvision==0.24.0 --index-url https://download.pytorch.org/whl30 |
There was a problem hiding this comment.
syntax: malformed PyTorch index URL - missing /cu130 after whl
| pip install torch==2.9.0 torchvision==0.24.0 --index-url https://download.pytorch.org/whl30 | |
| pip install torch==2.9.0 torchvision==0.24.0 --index-url https://download.pytorch.org/whl/cu130 |
| .. code-block:: bash | ||
|
|
||
| pip install -U torch==2.7.0 torchvision==0.22.0 --index-url https://download.pytorch.org/whl/cu128 | ||
| pip install -U torch==2.9.0 torchvision==0.24.0 --index-url https://download.pytorch.org/whl30 |
There was a problem hiding this comment.
syntax: malformed PyTorch index URL - missing /cu130 after whl
| pip install -U torch==2.9.0 torchvision==0.24.0 --index-url https://download.pytorch.org/whl30 | |
| pip install -U torch==2.9.0 torchvision==0.24.0 --index-url https://download.pytorch.org/whl/cu130 |
| .. code-block:: bash | ||
|
|
||
| pip install -U torch==2.7.0 torchvision==0.22.0 --index-url https://download.pytorch.org/whl/cu128 | ||
| pip install -U torch==2.9.0 torchvision==0.24.0 --index-url https://download.pytorch.org/whl30 |
There was a problem hiding this comment.
syntax: malformed PyTorch index URL - missing /cu130 after whl
| pip install -U torch==2.9.0 torchvision==0.24.0 --index-url https://download.pytorch.org/whl30 | |
| pip install -U torch==2.9.0 torchvision==0.24.0 --index-url https://download.pytorch.org/whl/cu130 |
| .. code-block:: bash | ||
|
|
||
| pip install -U torch==2.9.0 torchvision==0.24.0 --index-url https://download.pytorch.org/whl/cu130 | ||
| pip install -U torch==2.9.0 torchvision==0.24.0 --index-url https://download.pytorch.org/whl30 |
There was a problem hiding this comment.
syntax: malformed PyTorch index URL - missing /cu130 after whl
| pip install -U torch==2.9.0 torchvision==0.24.0 --index-url https://download.pytorch.org/whl30 | |
| pip install -U torch==2.9.0 torchvision==0.24.0 --index-url https://download.pytorch.org/whl/cu130 |
docs/source/setup/quickstart.rst
Outdated
| .. code-block:: bash | ||
|
|
||
| pip install -U torch==2.7.0 torchvision==0.22.0 --index-url https://download.pytorch.org/whl/cu128 | ||
| pip install -U torch==2.9.0 torchvision==0.24.0 --index-url https://download.pytorch.org/whl30 |
There was a problem hiding this comment.
syntax: malformed PyTorch index URL - missing /cu130 after whl
| pip install -U torch==2.9.0 torchvision==0.24.0 --index-url https://download.pytorch.org/whl30 | |
| pip install -U torch==2.9.0 torchvision==0.24.0 --index-url https://download.pytorch.org/whl/cu130 |
| import torch | ||
| from datetime import datetime | ||
|
|
||
| from rsl_rl.runners import DistillationRunner, OnPolicyRunner |
Signed-off-by: Kelly Guo <kellyguo123@hotmail.com>
Removed redundant environment variable export for ISAACSIM_ACCEPT_EULA. Signed-off-by: Kelly Guo <kellyguo123@hotmail.com>
Added dependencies to resolve pip errors. Signed-off-by: Kelly Guo <kellyguo123@hotmail.com>
Signed-off-by: Kelly Guo <kellyguo123@hotmail.com>
Removed unnecessary dependencies to resolve pip errors. Signed-off-by: Kelly Guo <kellyguo123@hotmail.com>
Signed-off-by: Kelly Guo <kellyguo123@hotmail.com>
Signed-off-by: Kelly Guo <kellyguo123@hotmail.com>
…into update-torch
9338da9
into
isaac-sim:feature/isaacsim-6-0
| isaac_sim_version = float(".".join(get_version()[2])) | ||
| if isaac_sim_version < 5: | ||
| logger.warning("[Compat] Isaac Sim < 5.0 does not support thread-local stage contexts. Skipping use_stage().") | ||
| yield # no-op | ||
| else: | ||
| if isaac_sim_version < 6: | ||
| # Set context in both modules to ensure all Isaac Sim subsystems see the correct stage | ||
| with stage_utils.use_stage(stage): | ||
| yield | ||
| else: | ||
| # Import both stage utils modules for Isaac Sim 5.0+ | ||
| import isaacsim.core.experimental.utils.stage as experimental_stage_utils | ||
|
|
||
| # Set context in both modules to ensure all Isaac Sim subsystems see the correct stage | ||
| with stage_utils.use_stage(stage): | ||
| with experimental_stage_utils.use_stage(stage): | ||
| yield |
There was a problem hiding this comment.
logic: inverted version comparison logic - code for Isaac Sim 6.0+ runs when version < 6, and code for Isaac Sim < 6.0 runs when version >= 6
| isaac_sim_version = float(".".join(get_version()[2])) | |
| if isaac_sim_version < 5: | |
| logger.warning("[Compat] Isaac Sim < 5.0 does not support thread-local stage contexts. Skipping use_stage().") | |
| yield # no-op | |
| else: | |
| if isaac_sim_version < 6: | |
| # Set context in both modules to ensure all Isaac Sim subsystems see the correct stage | |
| with stage_utils.use_stage(stage): | |
| yield | |
| else: | |
| # Import both stage utils modules for Isaac Sim 5.0+ | |
| import isaacsim.core.experimental.utils.stage as experimental_stage_utils | |
| # Set context in both modules to ensure all Isaac Sim subsystems see the correct stage | |
| with stage_utils.use_stage(stage): | |
| with experimental_stage_utils.use_stage(stage): | |
| yield | |
| isaac_sim_version = float(".".join(get_version()[2])) | |
| if isaac_sim_version < 6: | |
| # For Isaac Sim < 6.0, use single stage context | |
| with stage_utils.use_stage(stage): | |
| yield | |
| else: | |
| # Import both stage utils modules for Isaac Sim 6.0+ | |
| import isaacsim.core.experimental.utils.stage as experimental_stage_utils | |
| # Set context in both modules to ensure all Isaac Sim subsystems see the correct stage | |
| with stage_utils.use_stage(stage): | |
| with experimental_stage_utils.use_stage(stage): | |
| yield |
Description
Updates torch to 2.9.0, keeping cuda version as 12.8 for x86 and 13.0 for arm (for spark support).
In Isaac Sim 6.0, use_stage needs to be registered for both the regular isaacsim core extension and the core experimental extension
Type of change
Checklist
pre-commitchecks with./isaaclab.sh --formatconfig/extension.tomlfileCONTRIBUTORS.mdor my name already exists there