consistency_models
consistency_models copied to clipboard
How to run on a single linux server with multiple GPUs
Nice Job! I wonder how I can run the code on a single linux server with multiple GPUs. I can run the code on the server with one GPU by not using mpiexec. But what if I want to use multiple GPUs as nn.DataParallel?