mlcommons / training

Reference implementations of MLPerf™ training benchmarks
https://mlcommons.org/en/groups/training
Apache License 2.0
1.57k stars 548 forks source link

Running udnet3d on multiple GPUS #750

Open luiceur opened 3 days ago

luiceur commented 3 days ago

How can one run the udnet3d training on multiple GPUs?

The following doesn't work and it will launch multiple processes on a single GPU mpirun -np 4 ./run_and_time.sh