Unverified Commit 8aa39512 authored by Connor Baker's avatar Connor Baker Committed by GitHub
Browse files

Merge pull request #261656 from ConnorBaker/fix/torch-relax-cuda-version-requirements 

python3Packages.torch: relax CUDA version requirements
parents bb3f93ed fedb6672

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

+1 −1
Original line number Diff line number Diff line
@@ -111,7 +111,7 @@ let 
  brokenConditions = attrsets.filterAttrs (_: cond: cond) {

    "CUDA and ROCm are not mutually exclusive" = cudaSupport && rocmSupport;

    "CUDA is not targeting Linux" = cudaSupport && !stdenv.isLinux;

    "Unsupported CUDA version" = cudaSupport && (cudaPackages.cudaMajorVersion != "11");

    "Unsupported CUDA version" = cudaSupport && !(builtins.elem cudaPackages.cudaMajorVersion [ "11" "12" ]);

    "MPI cudatoolkit does not match cudaPackages.cudatoolkit" = MPISupport && cudaSupport && (mpi.cudatoolkit != cudaPackages.cudatoolkit);

    "Magma cudaPackages does not match cudaPackages" = cudaSupport && (magma.cudaPackages != cudaPackages);

  };