Cuda error while using YCB Dataset images with run_demo.py

NVlabs / FoundationPose

[CVPR 2024 Highlight] FoundationPose: Unified 6D Pose Estimation and Tracking of Novel Objects

https://nvlabs.github.io/FoundationPose/

Other

1.01k stars 112 forks source link

Open Sar-thak-3 opened 2 weeks ago

Sar-thak-3 commented 2 weeks ago

While using YCB Dataset on run_demo.py, it generates this error

RuntimeError: Cuda error: 2[cudaMalloc(&m_gpuPtr, bytes);

I changed get_depth(), get_color() and get_mask() according to the YCBReader but still the error persists.

wenbowen123 commented 2 weeks ago

what is your GPU memory?