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

Only link cuda driver librairies.

Browse Source
This commit is contained in:
Nicolas Patry 2024-09-13 16:50:29 +02:00
parent e898483db6
commit 3d7b81535a
No known key found for this signature in database
GPG Key ID: 64AF4752B2967863
1 changed files with 0 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -36,11 +36,6 @@ jobs:
run: | run: |
sudo mkdir -p /run/opengl-driver/lib sudo mkdir -p /run/opengl-driver/lib
sudo find /usr/lib . -name 'libcuda.so*' -exec ln -s {} /run/opengl-driver/lib/ \; sudo find /usr/lib . -name 'libcuda.so*' -exec ln -s {} /run/opengl-driver/lib/ \;
sudo find /usr/lib . -name 'libcudart.so*' -exec ln -s {} /run/opengl-driver/lib/ \;
sudo find /usr/lib . -name 'libcufft.so*' -exec ln -s {} /run/opengl-driver/lib/ \;
sudo find /usr/lib . -name 'libcublas.so*' -exec ln -s {} /run/opengl-driver/lib/ \;
sudo find /usr/lib . -name 'libcublasLt.so*' -exec ln -s {} /run/opengl-driver/lib/ \;
sudo find /usr/lib . -name 'libcurand.so*' -exec ln -s {} /run/opengl-driver/lib/ \;
sudo find /usr/lib . -name 'libnvidia-ml.so*' -exec ln -s {} /run/opengl-driver/lib/ \; sudo find /usr/lib . -name 'libnvidia-ml.so*' -exec ln -s {} /run/opengl-driver/lib/ \;
export LD_LIBRARY_PATH=/run/opengl-driver/lib export LD_LIBRARY_PATH=/run/opengl-driver/lib
nix develop .#test --command python -m pytest -svvx integration-tests/ nix develop .#test --command python -m pytest -svvx integration-tests/