How to set multiple GPUs

nqanh / affordance-net

AffordanceNet - Multiclass Instance Segmentation Framework - ICRA 2018

Other

121 stars 54 forks source link

How to set multiple GPUs #4

Open NKXLAB opened 6 years ago

NKXLAB commented 6 years ago

Good Job！ but How to set multiple GPUs？

nqanh commented 6 years ago

Our core caffe code will work with multple GPUs (see this), however you may have to modify some IO utils to make them work with multi-gpus. Unfortunately, I don't have access to multi-gpus at the moment to test it.