How to train multi models using one pc with multi GPUs?

WXinlong / SOLO

SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.

Other

1.71k stars 306 forks source link

Closed segust closed 3 years ago

segust commented 3 years ago

It's like you have no 'gpu-ids' in tools/train.py, so how to train multi models use one pc with multi GPUs?

segust commented 3 years ago

os.environ['CUDA_VISIBLE_DEVICES']='1, 2, 3'