diff --git a/pkgs/development/python-modules/torch/default.nix b/pkgs/development/python-modules/torch/default.nix index ba1da5844b17..09732e5e78e9 100644 --- a/pkgs/development/python-modules/torch/default.nix +++ b/pkgs/development/python-modules/torch/default.nix @@ -55,6 +55,7 @@ let inherit (cudaPackages) cudatoolkit cudaFlags cudnn nccl; in +assert cudaSupport -> stdenv.isLinux; assert cudaSupport -> (cudaPackages.cudaMajorVersion == "11"); # confirm that cudatoolkits are sync'd across dependencies