Skip to content

docs(install): document NCCL dev libraries for multi-GPU tensor parallelism - #90

Open
alogotron wants to merge 1 commit into
FlashML-org:mainfrom
alogotron:docs/nccl-tp-prereq
Open

docs(install): document NCCL dev libraries for multi-GPU tensor parallelism#90
alogotron wants to merge 1 commit into
FlashML-org:mainfrom
alogotron:docs/nccl-tp-prereq

Conversation

@alogotron

Copy link
Copy Markdown

Summary

Adds the NCCL development libraries as a documented prerequisite for --tensor-parallel-size > 1.

On a clean install (Ubuntu 24.04, PyPI wheels, CUDA 13 toolkit via NVIDIA apt repo), the first TP>1 launch fails during the JIT link step:

/usr/bin/ld: cannot find -lnccl

Resolved with:

sudo apt-get install libnccl2 libnccl-dev

install.md previously listed only driver r580+ / CUDA 13 and the nvcc requirement. Surfaced while testing TP=2 on 2x RTX 3090; also reported as a secondary finding in #29.

calvarado2004 added a commit to calvarado2004/FreeToken that referenced this pull request Aug 23, 2026
Implement the deployment feedback from the TP community testing: discover either an unversioned NCCL development link or a versioned runtime bundled with PyTorch, and fail with the searched paths before spawning ranks when neither exists.

Expose a dedicated distributed/rendezvous port while preserving the existing API-port-plus-one default. Wire the DeepSeek-V4 test harness to the explicit port it already checks, so its collision guard and the engine use the same endpoint.

Follow-up-to: FlashML-org#70

Builds-on: FlashML-org#90
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant