latent-diffusion icon indicating copy to clipboard operation
latent-diffusion copied to clipboard

cannot run on multi gpus

Open shencuifeng opened this issue 2 years ago • 3 comments

I start training with this command 'python main.py --base configs/autoencoder/vqmodel1.yaml -t --gpus 4,5' but I got this image everything works fine, steps in one epoch are halved, but only one gpu is in use, and only started one process. How to solve this problem?

shencuifeng avatar Dec 23 '22 09:12 shencuifeng

I start training with this command 'python main.py --base configs/autoencoder/vqmodel1.yaml -t --gpus 4,5' but I got this image everything works fine, steps in one epoch are halved, but only one gpu is in use, and only started one process. How to solve this problem?

have you found a solution yet? I am facing the same issue

OvO1111 avatar Dec 27 '23 04:12 OvO1111