From 7a5855ff01bbe2489c48d19e1d15cb9f257837b5 Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Tue, 17 Sep 2024 09:37:05 +0200 Subject: [PATCH] NCCL ? --- .github/workflows/nix_integration_tests.yaml | 2 ++ 1 file changed, 2 insertions(+) diff --git a/.github/workflows/nix_integration_tests.yaml b/.github/workflows/nix_integration_tests.yaml index 5c534b00..88ebc13d 100644 --- a/.github/workflows/nix_integration_tests.yaml +++ b/.github/workflows/nix_integration_tests.yaml @@ -34,6 +34,8 @@ jobs: run: | sudo mkdir -p /run/opengl-driver/lib/ export NVIDIA=/usr/lib/ + export NCCL_DEBUG=1 + export NCCL_P2P_DISABLE=1 sudo find $NVIDIA . -name 'libcuda.so*' -exec ln -s {} /run/opengl-driver/lib/ \; sudo find $NVIDIA . -name 'libnvidia-ml.so*' -exec ln -s {} /run/opengl-driver/lib/ \; nix develop .#test --command pytest -svvx integration-tests/ --release