search

drprojects / DeepViewAgg

[CVPR'22 Best Paper Finalist] Official PyTorch implementation of the method presented in "Learning Multi-View Aggregation In the Wild for Large-Scale 3D Semantic Segmentation"
Other
222 stars 24 forks source link

Is it possible to run with multiple GPU? #21

Closed Daniellli closed 1 year ago

Daniellli commented 1 year ago

Hi, Thank you for your kindly sharing codes. As the title shown , I am wondering whether such codebase supports multiple GPU run.

appreciate for your help.

drprojects commented 1 year ago

Hello @Daniellli, thanks for your interest in the project !

As of now, the code does not support multi-GPU. However, the current project is based on torch-points3d, which does support multi-GPU. Only, I did not build DeepViewAgg to make use of it. So I think it may be doable but I do not have time to work on this at the moment and cannot vouch for it .

If you decide to work on this, I would gladly accept a PR in this direction :wink:

drprojects commented 1 year ago

Closing this issue for now, feel free to open it again if need be