diff --git a/docs/design/mscclpp-dsl.md b/docs/design/mscclpp-dsl.md index cfb8072f..2bd7e162 100644 --- a/docs/design/mscclpp-dsl.md +++ b/docs/design/mscclpp-dsl.md @@ -12,7 +12,7 @@ pip install . ### Import mscclpp language module ```python -import mscclpp.language * +from mscclpp.language import * from mscclpp.language.types import ChannelType, ReplicationPolicy from mscclpp.language.collectives import AllGather diff --git a/docs/getting-started/quickstart.md b/docs/getting-started/quickstart.md index 1155934a..4403cad8 100644 --- a/docs/getting-started/quickstart.md +++ b/docs/getting-started/quickstart.md @@ -39,7 +39,7 @@ We provide docker images which package all prerequisites for MSCCL++. You can setup your dev environment with the following command. ```bash -$ docker run -it --privileged --net=host --ipc=host --gpus all ghcr.io/microsoft/mscclpp/mscclpp:base-dev-cuda12.2 mscclpp-dev bash +$ docker run -it --privileged --net=host --ipc=host --gpus all --name mscclpp-dev ghcr.io/microsoft/mscclpp/mscclpp:base-dev-cuda12.4 bash ``` See all available images [here](https://github.com/microsoft/mscclpp/pkgs/container/mscclpp%2Fmscclpp).