ddp support？

Open MrLihj opened this issue 8 months ago • 1 comments

Does this library not support multi-GPU training? The following error is reported:

RuntimeError: grid.options().device().is_cuda() INTERNAL ASSERT FAILED at "/tmp/pip-install-jec442jy/diso_2293b82a438d44a59a8feeffaaf14a16/src/pybind.cpp":45, please report a bug to PyTorch. grid must be a CUDA tensor.

Apr 21 '25 05:04 MrLihj

The library supports multi-GPU training, are you using the module as example here: https://github.com/SarahWeiii/diso?tab=readme-ov-file#batch-training ?

Apr 30 '25 21:04 SarahWeiii