MPI4cuML Performance on GPUs

Machine Specifications

CPU Model CPU Core Info Memory IB Card OS GPU CUDA
Intel Xeon Gold 6248 2x20 @ 2.5Ghz 384 GB Infiniband HDR (200 Gbps) Rocky Linux 8.5 NVIDIA V100 (4/Node) CUDA 11.2

K-Means Performance on Expanse

Model: K-Means ppn: 4
cuML K-Means on Expanse

Random Forests Performance on Expanse

Model: Random Forests ppn: 4
cuML Random Forests on Expanse

Linear Regression Performance on Expanse

Model: Linear Regression ppn: 4
cuML Linear Regression on Expanse

tSVD Performance on Expanse

Model: tSVD ppn: 4
cuML tSVD on Expanse

Nearest Neighbors Performance on Expanse

Model: Nearest Neighbors ppn: 4
cuML Nearest Neighbors on Expanse