mirror of
https://github.com/vale981/poetry2nix
synced 2025-03-05 17:21:39 -05:00
Apply suggestions from code review
This commit is contained in:
parent
8936c1c7af
commit
31f5bd3905
1 changed files with 1 additions and 3 deletions
|
@ -682,8 +682,6 @@ lib.composeManyExtensions [
|
|||
});
|
||||
|
||||
deepspeed = super.deepspeed.overridePythonAttrs (old: rec {
|
||||
# autoPatchelfIgnoreMissingDeps = true;
|
||||
|
||||
CUDA_HOME = pkgs.symlinkJoin {
|
||||
name = "deepspeed-cuda-home";
|
||||
paths = [
|
||||
|
@ -693,7 +691,7 @@ lib.composeManyExtensions [
|
|||
pkgs.cudaPackages.cuda_nvcc
|
||||
];
|
||||
};
|
||||
buildInputs = (old.buildInputs or [ ])++ [ self.setuptools ];
|
||||
buildInputs = old.buildInputs or [ ] ++ [ self.setuptools ];
|
||||
LD_LIBRARY_PATH = "${CUDA_HOME}/lib";
|
||||
preBuild = ''
|
||||
# Prevent the build from trying to access the default triton cache directory under /homeless-shelter
|
||||
|
|
Loading…
Add table
Reference in a new issue