python3Packages.torch: NCCL static library is in static output

NixOS · Sep 13, 2024 · 62796e8 · 62796e8
1 parent fed9d55
commit 62796e8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pkgs/development/python-modules/torch/default.nix b/pkgs/development/python-modules/torch/default.nix
@@ -478,7 +478,8 @@ buildPythonPackage rec {
       ++ lists.optionals (cudaPackages ? cudnn) [ cudnn ]
       ++ lists.optionals useSystemNccl [
         # Some platforms do not support NCCL (i.e., Jetson)
-        nccl # Provides nccl.h AND a static copy of NCCL!
+        (lib.getDev nccl)
+        (lib.getOutput "static" nccl)
       ]
       ++ lists.optionals (strings.versionOlder cudaVersion "11.8") [
         cuda_nvprof # <cuda_profiler_api.h>