diff --git a/.github/workflows/nix_integration_tests.yaml b/.github/workflows/nix_integration_tests.yaml
index b1bd5ea00..c5cd8e79a 100644
--- a/.github/workflows/nix_integration_tests.yaml
+++ b/.github/workflows/nix_integration_tests.yaml
@@ -32,7 +32,10 @@ jobs:
       run: nix develop .#test --command echo "Ok"
     - name: Python tests.
       run: |
-        export LD_LIBRARY_PATH=$(nix-build -E 'with import <nixpkgs> { config = { allowUnfree = true; }; }; linuxPackages.nvidia_x11')
+        sudo mkdir -p /run/opengl-driver/lib/
+        export NVIDIA=$(nix-build -E 'with import <nixpkgs> { config = { allowUnfree = true; }; }; linuxPackages.nvidia_x11')
+        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 -svv integration-tests/ --release
       env:
         HF_TOKEN: ${{ secrets.HF_TOKEN }}