search

szymanowiczs / splatter-image

Official implementation of `Splatter Image: Ultra-Fast Single-View 3D Reconstruction' CVPR 2024
https://szymanowiczs.github.io/splatter-image
BSD 3-Clause "New" or "Revised" License
795 stars 54 forks source link

Multi-GPU support #14

Closed thucz closed 7 months ago

thucz commented 7 months ago

Does the code support multi-gpu running? When I used the code in a 2-gpu machine. It seems that it only used one GPU

szymanowiczs commented 7 months ago

Yes - currently this only supports a single GPU because training and testing is fast enough to do experimentation with one GPU. Multi-GPU support could indeed make it quicker, but I don't anticipate adding that functionality soon so feel free to make a PR if you would like to.