We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 04efefa commit 5a3965eCopy full SHA for 5a3965e
1 file changed
pkgs/development/python-modules/torch/default.nix
@@ -273,7 +273,7 @@ in buildPythonPackage rec {
273
PYTORCH_BUILD_VERSION = version;
274
PYTORCH_BUILD_NUMBER = 0;
275
276
- USE_NCCL = setBool (cudaPackages ? nccl);
+ USE_NCCL = setBool (cudaSupport && cudaPackages ? nccl);
277
USE_SYSTEM_NCCL = setBool useSystemNccl; # don't build pytorch's third_party NCCL
278
USE_STATIC_NCCL = setBool useSystemNccl;
279
0 commit comments