No module named 'omni.isaac'

OK I found the solution

# This is the config used to launch simulation. 
CONFIG = {"renderer": "RayTracedLighting", "headless": args.headless, 
          "width": args.width, "height": args.height, "num_frames": args.num_frames}

simulation_app = SimulationApp(launch_config=CONFIG)


from omni.isaac.core.utils.extensions import enable_extension

enable_extension("omni.replicator.isaac")

from omni.replicator.isaac.scripts.writers import YCBVideoWriter
2 Likes