From ca1b22ba98eb882af99684d006e746183c94d73a Mon Sep 17 00:00:00 2001
From: Ivan Malison <IvanMalison@gmail.com>
Date: Tue, 31 Dec 2024 01:09:26 -0700
Subject: [PATCH] [NixOS] Wrap nvidia-cdi-hook with LD_LIBRARY_PATH setting

---
 nixos/nvidia-container-toolkit-overlay.nix | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/nixos/nvidia-container-toolkit-overlay.nix b/nixos/nvidia-container-toolkit-overlay.nix
index 93518cdc..2f2768ca 100644
--- a/nixos/nvidia-container-toolkit-overlay.nix
+++ b/nixos/nvidia-container-toolkit-overlay.nix
@@ -12,6 +12,7 @@ final: prev: {
     src = null;
     dontUnpack = true;
     dontPatchShell = true;
+    nativeBuildInputs = [ final.installShellFiles ] ++ final.lib.optionals final.stdenv.hostPlatform.isLinux [ final.makeWrapper ];
 
     buildInputs = [ final.strace ];
 
@@ -113,6 +114,7 @@ final: prev: {
 
       if [ -d wrapper-tools ]; then
       cp wrapper-tools/* $tools/bin/
+      wrapProgram $tools/bin/nvidia-cdi-hook --set LD_LIBRARY_PATH ${final.glibc}/lib
       fi
     '';