NCCL ?

2024-09-17 09:37:05 +02:00 · 2024-09-17 09:37:05 +02:00 · 7a5855ff01
parent fb7e8c8970
commit 7a5855ff01
1 changed files with 2 additions and 0 deletions
--- 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