updated paths

This commit is contained in:
alsaiduq-lab 2025-05-21 11:32:12 -06:00
parent 2e15174052
commit ea882cec1d

View file

@ -18,6 +18,7 @@
gccLibPath = "${pkgs.gcc-unwrapped.lib}/lib"; gccLibPath = "${pkgs.gcc-unwrapped.lib}/lib";
nvidiaLibPath = "${pkgs.linuxPackages.nvidia_x11}/lib"; nvidiaLibPath = "${pkgs.linuxPackages.nvidia_x11}/lib";
cudaLibPath = "${pkgs.cudatoolkit}/lib"; cudaLibPath = "${pkgs.cudatoolkit}/lib";
ldLibraryPath = "${gccLibPath}:${nvidiaLibPath}:${cudaLibPath}";
pythonEnv = pkgs.python311.buildEnv.override { pythonEnv = pkgs.python311.buildEnv.override {
extraLibs = with py; [ extraLibs = with py; [
@ -41,7 +42,7 @@
extraOutputsToInstall = ["out"]; extraOutputsToInstall = ["out"];
postBuild = '' postBuild = ''
wrapProgram $out/bin/python \ wrapProgram $out/bin/python \
--prefix LD_LIBRARY_PATH : "${gccLibPath}:${nvidiaLibPath}:${cudaLibPath}" --prefix LD_LIBRARY_PATH : "${ldLibraryPath}"
''; '';
}; };
@ -51,7 +52,8 @@
buildInputs = [pkgs.makeWrapper]; buildInputs = [pkgs.makeWrapper];
postBuild = '' postBuild = ''
wrapProgram $out/bin/uv \ wrapProgram $out/bin/uv \
--prefix LD_LIBRARY_PATH : "${gccLibPath}:${nvidiaLibPath}:${cudaLibPath}" --prefix LD_LIBRARY_PATH : "${ldLibraryPath}" \
--set PYTHONPATH ""
''; '';
}; };
in { in {
@ -69,7 +71,7 @@ in {
]; ];
environment.variables = { environment.variables = {
PIP_CONFIG_FILE = "${pipConf}"; PIP_CONFIG_FILE = "${pipConf}";
LD_LIBRARY_PATH = lib.mkForce "${gccLibPath}:${nvidiaLibPath}:${cudaLibPath}"; LD_LIBRARY_PATH = lib.mkForce "${ldLibraryPath}";
}; };
}; };
} }