updated paths
This commit is contained in:
parent
2e15174052
commit
ea882cec1d
1 changed files with 5 additions and 3 deletions
|
|
@ -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}";
|
||||||
};
|
};
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue