How to train on single GPU

[Chen65010445]

For I don't have multi-GPUS, I changed the GPUS = (0,0,0,0), but this error came RuntimeError: inputs must be on unique devices. There is no doubt telling that this network working on more than one GPUS. So how do I change the code to run them on single GPU?

[LvShuaiChao]

Find your corresponding yaml file, and then change its GPUs number to a separate 0.

[piyushchauhan]

Remove the GPUS: from the yaml file that you are using. By default GPU is set to 0.
Refer:

HRNet-Image-Classification/lib/config/default.py

Line 23 in f760c98

_C.GPUS = (0,)

[HRHLALALA]

@piyushchauhan @LvShuaiChao Hi, have you done any other operations? I have just simply changed the GPUs but got the error: RuntimeError: Default process group has not been initialized, please make sure to call init_process_group. Do you know how to solve it?

[Mac-007]

In experiments/*.yaml file, change it to -> GPUS: (0,)