From 98f2241a88a5158b5ae99d16ce8fc494070d68a2 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Sat, 14 Sep 2024 23:29:12 +0200
Subject: [PATCH] Put back libnvidia-ml

---
 .github/workflows/nix_integration_tests.yaml | 1 +
 flake.nix                                    | 1 -
 2 files changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/nix_integration_tests.yaml b/.github/workflows/nix_integration_tests.yaml
index 7e8ca8b62..239f56756 100644
--- a/.github/workflows/nix_integration_tests.yaml
+++ b/.github/workflows/nix_integration_tests.yaml
@@ -34,6 +34,7 @@ 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 'libnvidia-ml.so*' -exec ln -s {} /run/opengl-driver/lib/ \;
 
         nix develop .#test --command pytest -svv integration-tests/ --release
       env:
diff --git a/flake.nix b/flake.nix
index 45c28637d..674db0d82 100644
--- a/flake.nix
+++ b/flake.nix
@@ -95,7 +95,6 @@
                 cargo
                 rustfmt
                 clippy
-                cudaPackages.cudatoolkit
               ]
               ++ (with python3.pkgs; [
                 docker