Skip to content

Instantly share code, notes, and snippets.

@huseinzol05

huseinzol05/install.sh

Created June 25, 2024 07:30
Show Gist options
  • Save huseinzol05/04f93dfb57c463eb968f2162761e9540 to your computer and use it in GitHub Desktop.
Save huseinzol05/04f93dfb57c463eb968f2162761e9540 to your computer and use it in GitHub Desktop.
Download ZIP
NCCL test OpenMPI
Raw
install.sh
git clone https://github.com/nvidia/nccl-tests && cd nccl-tests
sudo apt-get install openmpi-bin openmpi-common libopenmpi-dev -y
make MPI=1 MPI_HOME=/usr/lib/x86_64-linux-gnu/openmpi
./build/all_reduce_perf -b 8 -e 128M -f 2 -g 4
mpirun -np 10 -H 10.224.0.47,10.224.0.70 ./build/all_reduce_perf -b 8 -e 128M -f 2 -g 4
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment