Unverified Commit ee8921a1 authored by Weijia Wang's avatar Weijia Wang Committed by GitHub
Browse files

Merge pull request #232480 from wegank/cudatoolkit-eval-2 

python310Packages.torchWithCuda: add platform check
parents b7783786 9775ce1b

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

+1 −0
Original line number Diff line number Diff line
@@ -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