yenchenlin / nerf-pytorch

A PyTorch implementation of NeRF (Neural Radiance Fields) that reproduces the results.
MIT License
5.37k stars 1.04k forks source link

Pytorch 1.12.0 issue #72

Closed Yuchen-Song closed 2 years ago

Yuchen-Song commented 2 years ago

A kind reminder: do not use pytorch 1.12.0 for this project, which would throw an error:

_assert not step_t.is_cuda, "If capturable=False, state_steps should not be CUDA tensors." AssertionError: If capturable=False, statesteps should not be CUDA tensors.

I tried with torch 1.11.0, then it works.