fix: IS 4.1.0 + Blackwell GPU compatibility fixes

- Add scipy Rotation scalar_first monkey-patch for older scipy (<1.11)
- Fix SimulationApp import to support both IS 4.x and 5.x
- Reuse task object across reset() calls to prevent duplicate prims
- Add _scene_initialized guard in set_up_scene() for repeated resets
- Cache arena_file_path to survive task_cfg.pop()
- Clean up collision groups before re-creating
- Switch to PathTracing renderer for clean output on Blackwell GPU

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Tangger
2026-04-01 22:37:58 +08:00
parent 2a0a21f2c8
commit f338199bcb
6 changed files with 64 additions and 3 deletions

View File

@@ -45,6 +45,7 @@ class SimBoxDualWorkFlow(NimbusWorkFlow):
self.scene_info = scene_info
self.step_replay = False
self.random_seed = random_seed
self._arena_file_path = None
super().__init__(world, task_cfg_path)
def parse_task_cfgs(self, task_cfg_path: str) -> list:
@@ -76,7 +77,8 @@ class SimBoxDualWorkFlow(NimbusWorkFlow):
set_camera_view(eye=[1.3, 0.7, 2.7], target=[0.0, 0, 1.5], camera_prim_path="/OmniverseKit_Persp")
# Modify config
arena_file_path = self.task_cfg.get("arena_file", None)
arena_file_path = self.task_cfg.get("arena_file", None) or self._arena_file_path
self._arena_file_path = arena_file_path
with open(arena_file_path, "r", encoding="utf-8") as arena_file:
arena = yaml.load(arena_file, Loader=Loader)
@@ -113,9 +115,13 @@ class SimBoxDualWorkFlow(NimbusWorkFlow):
physx_interface = acquire_physx_interface()
physx_interface.overwrite_gpu_setting(1)
self.task = get_task_cls(self.task_cfg["task"])(self.task_cfg)
if not hasattr(self, "task") or self.task is None:
self.task = get_task_cls(self.task_cfg["task"])(self.task_cfg)
else:
self.task.cfg = self.task_cfg
self.stage = self.world.stage
self.stage.SetDefaultPrim(self.stage.GetPrimAtPath("/World"))
self.world._current_tasks.clear()
self.world.add_task(self.task)
# # Add hidden ground plane for physics simulation