csxmli2016 / DFDNet

Blind Face Restoration via Deep Multi-scale Component Dictionaries (ECCV 2020)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Multiple gpus?

pointave opened this issue · 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.