btx0424 / OmniDrones

https://omnidrones.readthedocs.io/
MIT License
147 stars 25 forks source link

Question about test_rate_controller.py #73

Open Omikaa opened 4 months ago

Omikaa commented 4 months ago

Thanks for your work. Now I am trying to run test_rate_controller.py in example folder.But there is a issue about CUDA.Can you provide me some advices? Details: Traceback (most recent call last): File "/home/omika/Omniverse/OmniDrones/examples/test_rate_controller.py", line 113, in main frames_vis.append(camera_vis.get_images().cpu()) File "/home/omika/Omniverse/OmniDrones/omni_drones/sensors/camera.py", line 140, in get_images img_tensor = wp.to_torch(v.get_data(device=self.device)) File "/home/omika/Omniverse/Omniverse_Library/isaac_sim-2023.1.1/extscache/omni.warp.core-1.0.0-beta.2/warp/torch.py", line 218, in to_torch t = torch.as_tensor(a, device=device_to_torch(a.device)) RuntimeError: The specified pointer resides on host memory and is not registered with any CUDA device.

And I have checked that torch.py in omni.warp. image I am sure that a.device is cude:0