torch fix, fixes #230 (#237)

* torch fix, fixes #230

Co-authored-by: Tom Sydney Kerckhove <syd@cs-syd.eu>
This commit is contained in:
Tom Sydney Kerckhove 2021-01-30 19:56:39 +01:00 committed by GitHub
parent ee478fd8d4
commit ba43d90b78
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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
];
})
)