Skip to content

Commit

Permalink
check libnccl instead of nccl to be more reliable
Browse files Browse the repository at this point in the history
  • Loading branch information
dengl11 committed Aug 14, 2024
1 parent 4c84bc7 commit e978e85
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -200,7 +200,7 @@ else
# Detect if running on macOS or Linux
ifeq ($(SHELL_UNAME), Darwin)
$(info ✗ Multi-GPU on CUDA on Darwin is not supported, skipping NCCL support)
else ifeq ($(shell dpkg -l | grep -q nccl && echo "exists"), exists)
else ifeq ($(shell dpkg -l | grep -q libnccl && echo "exists"), exists)
$(info ✓ NCCL found, OK to train with multiple GPUs)
NVCC_FLAGS += -DMULTI_GPU
NVCC_LDLIBS += -lnccl
Expand Down

0 comments on commit e978e85

Please sign in to comment.