DFDNet icon indicating copy to clipboard operation
DFDNet copied to clipboard

Multiple gpus?

Open pointave opened this issue 3 years ago • 0 comments

I get this error when trying --gpu_ids 0,1 "There is an imbalance between your gpus. You may want to exclude gpu 1 which has less than 75% of the memory or cores of gpu 0. You can do so by setting the device_ids argument to DataParallel, or by setting the cuda_visible_devices environment variable. warnings.warn(imbalance_warn.format(device_ids[min_pos], device_ids[max_pos]))"

I'm using a 2080 ti and a 1080 on windows 10 and would like to see if adding a second gpu would speed up the extractions.

Also if anyone knows how to run step4 on gpu, would greatly appreciate hearing from you.

pointave avatar May 04 '21 18:05 pointave