search

HealthML / self-supervised-3d-tasks

Apache License 2.0
187 stars 40 forks source link

Can we run this code on Multigpu Setting? #8

Closed ramchandracheke closed 3 years ago

ramchandracheke commented 3 years ago

Hi, I am running this code on single GPU. However, it is utilizing all memory. Can we run this on multiple gpu? Thanks,

aihamtaleb commented 3 years ago

Hi, Our current codebase doesn't support such a request, unfortunately. However, extending this in your clone should be straightforward, according to the following info: https://datascience.stackexchange.com/a/25737

Please check it out, and give it a try. If this works you're welcome to contribute a running script to our repo with a Pull Request.

Cheers

ramchandracheke commented 3 years ago

Thanks @aihamtaleb

Sure I will try and will update you.

Thanks,