mirror of
https://github.com/vale981/poetry2nix
synced 2025-03-05 09:11:39 -05:00
* torch fix, fixes #230 Co-authored-by: Tom Sydney Kerckhove <syd@cs-syd.eu>
This commit is contained in:
parent
ee478fd8d4
commit
ba43d90b78
1 changed files with 5 additions and 3 deletions
|
@ -1184,14 +1184,16 @@ self: super:
|
|||
# is explicitly disabled with USE_CUDA=0.
|
||||
find $out -name "*.so" -exec ${pkgs.patchelf}/bin/patchelf --remove-needed libcuda.so.1 {} \;
|
||||
'';
|
||||
buildInputs = old.buildInputs ++ lib.optionals enableCuda [
|
||||
buildInputs = (old.buildInputs or [ ])
|
||||
++ [ self.typing-extensions ]
|
||||
++ lib.optionals enableCuda [
|
||||
pkgs.linuxPackages.nvidia_x11
|
||||
pkgs.nccl.dev
|
||||
pkgs.nccl.out
|
||||
];
|
||||
propagatedBuildInputs = [
|
||||
super.numpy
|
||||
super.future
|
||||
self.numpy
|
||||
self.future
|
||||
];
|
||||
})
|
||||
)
|
||||
|
|
Loading…
Add table
Reference in a new issue