small-dataset-image-generation icon indicating copy to clipboard operation
small-dataset-image-generation copied to clipboard

Published 20 hours ago • verified publisher icon nogu-atsu

How to run on multiple gpu?

Open datduong opened this issue 4 years ago • 0 comments

Hi, the command "--gpu" specifies only the gpu to be used (for example, used the 2nd gpu instead of the first). How can I modify the code to run on multiple gpus? Thanks.

For me, using mpirun python ./train.py --config_path configs/default.yml does not automatically detect that there are more than 1 gpus available.

datduong avatar Jan 15 '21 04:01 datduong