This commit is contained in:
alsaiduq-lab 2025-05-22 06:50:56 -06:00
parent 2f50e92afe
commit b22792614e
9 changed files with 49 additions and 29 deletions

View file

@ -18,7 +18,8 @@
gccLibPath = "${pkgs.gcc-unwrapped.lib}/lib";
nvidiaLibPath = "${pkgs.linuxPackages.nvidia_x11}/lib";
cudaLibPath = "${pkgs.cudatoolkit}/lib";
ldLibraryPath = "${gccLibPath}:${nvidiaLibPath}:${cudaLibPath}";
glvndLibPath = "${pkgs.libglvnd}/lib";
ldLibraryPath = "${gccLibPath}:${nvidiaLibPath}:${cudaLibPath}:${glvndLibPath}";
pythonEnv = pkgs.python311.buildEnv.override {
extraLibs = with py; [
@ -36,6 +37,7 @@
jedi
libcst
wheel
jupyterlab
];
extraOutputsToInstall = ["out"];
postBuild = ''
@ -69,7 +71,6 @@ in {
];
environment.variables = {
PIP_CONFIG_FILE = "${pipConf}";
LD_LIBRARY_PATH = lib.mkForce "${ldLibraryPath}";
};
};
}