Skip to content

Cleanup External NCCL detection process#2853

Merged
soumith merged 3 commits intomasterfrom
ncclbuild
Sep 25, 2017
Merged

Cleanup External NCCL detection process#2853
soumith merged 3 commits intomasterfrom
ncclbuild

Conversation

@soumith
Copy link
Collaborator

@soumith soumith commented Sep 25, 2017

Introduces a consistent and cleaner way of detecting external NCCL, similar to CuDNN detection.

NCCL_ROOT_DIR="/foo/my_dir" python setup.py install finds nccl.h in /foo/my_dir/include and libnccl* in /foo/my_dir/lib or /foo/my_dir/lib64.

Alternatively, NCCL_LIB_DIR="/foo/my_dir/lib" python setup.py install finds nccl.h in /foo/my_dir/include and libnccl* in /foo/my_dir/lib

Also fixes a small build_libs.sh arg parsing bug.

Fixes #2553 and #2375

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.

external nccl breaks gloo compilation

1 participant