huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-05-04 00:32:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

NCCL attempts

Browse Source
This commit is contained in:
Nicolas Patry 2024-09-17 09:43:28 +02:00
parent 7a5855ff01
commit c859663f98
No known key found for this signature in database
GPG Key ID: 690EBED7128A9DC8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/nix_integration_tests.yaml vendored
View File

@ -34,7 +34,7 @@ jobs:
run: | run: |
sudo mkdir -p /run/opengl-driver/lib/ sudo mkdir -p /run/opengl-driver/lib/
export NVIDIA=/usr/lib/ export NVIDIA=/usr/lib/
export NCCL_DEBUG=1 export NCCL_DEBUG=INFO
export NCCL_P2P_DISABLE=1 export NCCL_P2P_DISABLE=1
sudo find $NVIDIA . -name 'libcuda.so*' -exec ln -s {} /run/opengl-driver/lib/ \; 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/ \; sudo find $NVIDIA . -name 'libnvidia-ml.so*' -exec ln -s {} /run/opengl-driver/lib/ \;