microsoft / otdd

Optimal Transport Dataset Distance
MIT License
151 stars 48 forks source link

How to use multiple GPUs for calculation? #26

Open Jackbrocp opened 2 years ago

Jackbrocp commented 2 years ago

When I use cpu for calculation, the time cost is so high. So I want to use multiple gpus for calculation. But compared with cpu, even with the same 'maxsample' setting, the memory cost of gpus increases dramatically and easily comes to ' CUDA: out of memory'. Is it possible to speed up the calculation with GPUs?