Skip to content

Commit

Permalink
Update quickstart.md (#314)
Browse files Browse the repository at this point in the history
Updating the docker image name tag and the python benchmark path.
  • Loading branch information
angelica-moreira authored Jun 19, 2024
1 parent 93ed8e1 commit 34f4d9d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions docs/quickstart.md
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ $ CXX=/path/to/hipcc python -m pip install .
Our base image installs all prerequisites for MSCCL++.

```bash
$ docker pull ghcr.io/microsoft/mscclpp/mscclpp:base-cuda12.3
$ docker pull ghcr.io/microsoft/mscclpp/mscclpp:base-dev-cuda12.3
```

See all available images [here](https://github.com/microsoft/mscclpp/pkgs/container/mscclpp%2Fmscclpp).
Expand Down Expand Up @@ -106,7 +106,7 @@ $ mpirun -np 16 -npernode 8 -hostfile hostfile ./test/mp_unit_tests -ip_port 10.
```bash
# Choose `requirements_*.txt` according to your CUDA/ROCm version.
$ python3 -m pip install -r ./python/requirements_cuda12.txt
$ mpirun -tag-output -np 8 python3 ./python/benchmark/allreduce_bench.py
$ mpirun -tag-output -np 8 python3 ./python/mscclpp_benchmark/allreduce_bench.py
```

### C++ Benchmark (mscclpp-test)
Expand Down

0 comments on commit 34f4d9d

Please sign in to comment.