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

NCCL ?

Browse Source
This commit is contained in:
Nicolas Patry 2024-09-17 09:37:05 +02:00
parent fb7e8c8970
commit 7a5855ff01
No known key found for this signature in database
GPG Key ID: 690EBED7128A9DC8
1 changed files with 2 additions and 0 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,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