search

PJLab-ADG / neuralsim

neuralsim: 3D surface reconstruction and simulation based on 3D neural rendering.
MIT License
590 stars 31 forks source link

Support for PyTorch 2.0 #5

Closed lxxue closed 1 year ago

lxxue commented 1 year ago

Hi Jianfei,

Thanks for sharing the great work. I have a workstation with a 4090 which requires at least cuda 11.8. For cuda 11.8, pytorch only has wheels with versions >= 2.0. I saw you were doing some fixes for pytorch 2.0. I wonder if it is ready so I can start playing with the neuralsim repo?

lxxue commented 1 year ago

Update: I cloned the newest nr3d_lib with commit b4c5d81. I experimented with scan 1 in Replica and results look good.

Sorry I just noticed the issue #4 so this one can be closed.

ventusff commented 1 year ago

:) Glad it works.