diff --git a/.github/workflows/nix_integration_tests.yaml b/.github/workflows/nix_integration_tests.yaml index 54051afe..7985c4f6 100644 --- a/.github/workflows/nix_integration_tests.yaml +++ b/.github/workflows/nix_integration_tests.yaml @@ -38,6 +38,12 @@ jobs: run: | 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 '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/ \; export LD_LIBRARY_PATH=/run/opengl-driver/lib nix develop .#test --command python -m pytest -svvx integration-tests/ env: