robo-alex / gs-dynamics

Code for "Dynamic 3D Gaussian Tracking for Graph-Based Neural Dynamics Modeling" (CoRL 2024)
https://gs-dynamics.github.io
33 stars 5 forks source link

tran_gs stuck at Renderer #3

Open YESAndy opened 1 week ago

YESAndy commented 1 week ago

Hi,

Thank you very much for releasing your work! I was trying to rerun the gs training but stuck for a long time (~30 mins) when running

im, radius, _, = Renderer(raster_settings=curr_data['cam'])(**rendervar) from train_utils.py

I suspected that it may be because of the installation configuration problem about CUDA. So I tried to reinstall the diff-gaussian-rasterization-w-depth package with CUDA_HOME=/path/to/cuda. But the problem still existed.

I checked the source code of the Renderer which is generated by python binding codes from C++. I have zero idea about how to debug with C++ ;)

Do you have any suggestion about the reason why it stuck? Thank you!

YESAndy commented 1 week ago

problem solved. it is because the mismatch between nvidia driver and torch version:) good job nvidia haha

robo-alex commented 3 days ago

This is great Andy! Thank you for your interest and please do let us know if you have further questions