Merge pull request #273630 from NixOS/backport-273594-to-release-23.11

[Backport release-23.11] python3Packages.torch: gate NCCL with cudaSupport
NixOS · Dec 11, 2023 · c11b7c2 · c11b7c2
2 parents e3af581 + 3ce7d44
commit c11b7c2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkgs/development/python-modules/torch/default.nix b/pkgs/development/python-modules/torch/default.nix
@@ -268,7 +268,7 @@ in buildPythonPackage rec {
   PYTORCH_BUILD_VERSION = version;
   PYTORCH_BUILD_NUMBER = 0;
 
-  USE_NCCL = setBool (cudaPackages ? nccl);
+  USE_NCCL = setBool (cudaSupport && cudaPackages ? nccl);
   USE_SYSTEM_NCCL = setBool useSystemNccl;                  # don't build pytorch's third_party NCCL
   USE_STATIC_NCCL = setBool useSystemNccl;