What kind of PC is needed to work with this program? about zipnerf-pytorch HOT 5 CLOSED

Hi, this repo is an implementation of zipnerf, which is used for 3D reconstruction. A machine with a GPU is needed.

from zipnerf-pytorch.

Hi, this repo is an implementation of zipnerf, which is used for 3D reconstruction. A machine with a GPU is needed.

Hi, thanks for answering, but which card is suitable for use?
I have a palit 3060oc 12 gb

from zipnerf-pytorch.

In zipnerf paper, they train it with 8 NVIDIA Tesla V100, and I use 8 NVIDIA RTX A6000. But a 3060 is enough if you can install the requirements.txt successfully, what you need to do is to decrease the batch size to make it work without OutOfMemory error.

from zipnerf-pytorch.

In zipnerf paper, they train it with 8 NVIDIA Tesla V100, and I use 8 NVIDIA RTX A6000. But a 3060 is enough if you can install the requirements.txt successfully, what you need to do is to decrease the batch size to make it work without OutOfMemory error.

Great , Thank you very much for the answer !
I will try when I get home

from zipnerf-pytorch.

@SuLvXiangXin Hi I do have a question about batch_size, in config.py we have batch_size=2 ** 16, but that gets divided by world_size, so when you trained with 8xA6000, you would have 2*13 rays per GPU(which uses about 8GB VRAM per GPU)? Could you confirm that's the setup for 360v2 dataset? Thanks

from zipnerf-pytorch.