Skip to content

Commit 5a3965e

Browse files
author
Connor Baker
committed
python3Packages.torch: gate NCCL with cudaSupport
1 parent 04efefa commit 5a3965e

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

pkgs/development/python-modules/torch/default.nix

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -273,7 +273,7 @@ in buildPythonPackage rec {
273273
PYTORCH_BUILD_VERSION = version;
274274
PYTORCH_BUILD_NUMBER = 0;
275275

276-
USE_NCCL = setBool (cudaPackages ? nccl);
276+
USE_NCCL = setBool (cudaSupport && cudaPackages ? nccl);
277277
USE_SYSTEM_NCCL = setBool useSystemNccl; # don't build pytorch's third_party NCCL
278278
USE_STATIC_NCCL = setBool useSystemNccl;
279279

0 commit comments

Comments
 (0)