Can not locate "omni.isaac.orbit"

[RuiqiZhang99]

Hi, I meet an issue after install all the package, could you help me to solve it? Version: Ubuntu20.04 IsaacSim-2023.1.0-hotfix.1 Python-3.10 After I install all the packages as tutorial and run train.py with my wandb id, it prints the config information and then a bug comes:

[2024-02-07 23:18:13,254][wandb][DEBUG] - no default config file found in config-defaults.yaml
task:
  env:
    num_envs: 4096
    env_spacing: 8
    max_episode_length: 500
  sim:
    dt: 0.016
    substeps: 1
    gravity:
    - 0
    - 0
    - -9.81
    replicate_physics: false
    use_flatcache: true
    use_gpu_pipeline: true
    device: cuda:0
    solver_type: 1
    use_gpu: true
    bounce_threshold_velocity: 0.2
    friction_offset_threshold: 0.04
    friction_correlation_distance: 0.025
    enable_stabilization: true
    gpu_max_rigid_contact_count: 524288
    gpu_max_rigid_patch_count: 163840
    gpu_found_lost_pairs_capacity: 4194304
    gpu_found_lost_aggregate_pairs_capacity: 33554432
    gpu_total_aggregate_pairs_capacity: 4194304
    gpu_max_soft_body_contacts: 1048576
    gpu_max_particle_contacts: 1048576
    gpu_heap_capacity: 67108864
    gpu_temp_buffer_capacity: 16777216
    gpu_max_num_partitions: 8
  name: Hover
  drone_model: Firefly
  force_sensor: false
  time_encoding: true
  reward_effort_weight: 0.1
  reward_action_smoothness_weight: 0.0
  reward_motion_smoothness_weight: 0.0
  reward_distance_scale: 1.2
  action_transform: null
algo:
  name: ppo
  train_every: 32
  ppo_epochs: 4
  num_minibatches: 16
  priv_actor: false
  priv_critic: false
  checkpoint_path: null
headless: true
sim:
  dt: 0.016
  substeps: 1
  gravity:
  - 0
  - 0
  - -9.81
  replicate_physics: false
  use_flatcache: true
  use_gpu_pipeline: true
  device: cuda:0
  solver_type: 1
  use_gpu: true
  bounce_threshold_velocity: 0.2
  friction_offset_threshold: 0.04
  friction_correlation_distance: 0.025
  enable_stabilization: true
  gpu_max_rigid_contact_count: 524288
  gpu_max_rigid_patch_count: 163840
  gpu_found_lost_pairs_capacity: 4194304
  gpu_found_lost_aggregate_pairs_capacity: 33554432
  gpu_total_aggregate_pairs_capacity: 4194304
  gpu_max_soft_body_contacts: 1048576
  gpu_max_particle_contacts: 1048576
  gpu_heap_capacity: 67108864
  gpu_temp_buffer_capacity: 16777216
  gpu_max_num_partitions: 8
env:
  num_envs: 4096
  env_spacing: 8
  max_episode_length: 500
total_frames: 150000000
max_iters: -1
eval_interval: -1
save_interval: -1
seed: 0
viewer:
  resolution:
  - 960
  - 720
  eye:
  - 8
  - 0.0
  - 6.0
  lookat:
  - 0.0
  - 0.0
  - 1.0
wandb:
  group: Hover
  run_name: Hover-ppo
  job_type: train
  entity: afc30efa2dd33891947c0ec7035bbe2f085d9243
  project: omnidrones
  mode: disabled
  run_id: null
  monitor_gym: true
  tags: null

Error executing job with overrides: ['algo=ppo', 'wandb.entity=afc30efa2dd33891947c0ec7035bbe2f085d9243']
Traceback (most recent call last):
  File "/home/rich/OmniDrones/scripts/train.py", line 48, in main
    from omni_drones.envs.isaac_env import IsaacEnv
  File "/home/rich/OmniDrones/omni_drones/envs/__init__.py", line 32, in <module>
    from .pinball import Pinball
  File "/home/rich/OmniDrones/omni_drones/envs/pinball.py", line 43, in <module>
    from omni.isaac.orbit.sensors import ContactSensorCfg, ContactSensor
ModuleNotFoundError: No module named 'omni.isaac.orbit'

Set the environment variable HYDRA_FULL_ERROR=1 for a complete stack trace.

Is the omni.isaac.orbit an extension of isaacsim? Do we need to download and install it additionally? THANKS