jolheiser/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #273594 from ConnorBaker/fix/pytorch-fix-gate-nccl-with-cudaSupport

Browse Source 
python3Packages.torch: gate NCCL with cudaSupport
This commit is contained in:
Connor Baker 2023-12-11 15:24:53 -05:00 committed by GitHub
commit 3c1873efa9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pkgs/development/python-modules/torch/default.nix
View File

@ -273,7 +273,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;