nerfstudio-project / nerfstudio

A collaboration friendly studio for NeRFs
https://docs.nerf.studio
Apache License 2.0
8.89k stars 1.18k forks source link

Update pytorch in Dockerfile to 2.1.2 #3268

Closed Zunhammer closed 4 days ago

Zunhammer commented 5 days ago

Update pytorch to fix gsplat incompatibilities with versions >= 1.0.0 Additionally, version pinning of tinycudann as 1.6 was removed as it is not compatible with the new pytorch setuptools is now fixed to 69.5.1 also because of incompatibilities Currently, numpy 2 creates an error but as others are already working on fixing this in nerfstudio I'm skipping it in the Dockerfile. In case you need to fix it now please pin the version in nerfstudio dependencies or add a line in the Dockerfile before the ns-install-cli command: RUN python3.10 -m pip install --no-cache-dir numpy==1.26.4

Fixes #3235