D-NeRF CUDA out of memory
m-pabst opened this issue · comments
Hi,
my specs
Windows 11
Cuda toolkit 11.8
pytorch: 2.0.0+cu118
RTX 3070 8 GB, GDDR6
Tried to run:
python main_dnerf.py data/dnerf/jumpingjacks --workspace trial_dnerf_basis_jumpingjacks --fp16 --cuda_ray --bound 1.0 --scale 0.8 --dt_gamma 0 --basis
error:
renderer.py
line 540, in update_extra_state
self.density_grid[valid_mask] = torch.maximum(self.density_grid[valid_mask] * decay, tmp_grid[valid_mask])
torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 3.00 GiB (GPU 0; 8.00 GiB total capacity; 1.83 GiB already allocated; 1.64 GiB free; 4.53 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation.
Any advice how to solve this?
Thanks.