Move cudnn pin to tests

Signed-off-by: Michał Górny <[email protected]>

Author: mgorny

recipe/meta.yaml

@@ -121,9 +121,7 @@ requirements:
     - rsync     # [unix]
   host:
     # GPU requirements
-    # cudnn pinned over segfaults
-    # https://github.com/conda-forge/pytorch-cpu-feedstock/pull/411
-    - cudnn <9.11                     # [cuda_compiler_version != "None"]
+    - cudnn                           # [cuda_compiler_version != "None"]
     - nccl                            # [cuda_compiler_version != "None" and linux]
     - magma                           # [cuda_compiler_version != "None"]
     - cuda-version {{ cuda_compiler_version }}  # [cuda_compiler_version != "None"]
@@ -174,7 +172,7 @@ requirements:
     - zlib
   run:
     # GPU requirements without run_exports
-    - cudnn <9.11                       # [cuda_compiler_version != "None"]
+    - cudnn                             # [cuda_compiler_version != "None"]
     - intel-openmp {{ mkl }}            # [win]
     - libblas * *{{ blas_impl }}        # [blas_impl == "mkl"]
   run_constrained:
@@ -277,7 +275,7 @@ outputs:
       host:
         - {{ pin_subpackage('libtorch', exact=True) }}
         # GPU requirements
-        - cudnn <9.11                     # [cuda_compiler_version != "None"]
+        - cudnn                           # [cuda_compiler_version != "None"]
         - nccl                            # [cuda_compiler_version != "None" and linux]
         - cuda-version {{ cuda_compiler_version }}  # [cuda_compiler_version != "None"]
         - nvtx-c                          # [cuda_compiler_version != "None"]
@@ -332,7 +330,7 @@ outputs:
         - libblas * *{{ blas_impl }}        # [blas_impl == "mkl"]
         - nomkl                             # [blas_impl != "mkl"]
         # GPU requirements without run_exports
-        - cudnn <9.11                       # [cuda_compiler_version != "None"]
+        - cudnn                             # [cuda_compiler_version != "None"]
         - triton =={{ triton }}             # [cuda_compiler_version != "None" and not win]
         # avoid that people without GPUs needlessly download ~0.5-1GB
         - __cuda                            # [cuda_compiler_version != "None"]
@@ -386,6 +384,10 @@ outputs:
         - cuda-nvrtc-dev            # [cuda_compiler_version != "None"]
         - nvtx-c                    # [cuda_compiler_version != "None"]
         - pybind11
+        # cudnn >=9.11 removed support older GPUs that are still supported by CUDA 12
+        # pin tests since the CI uses sm70 that is no longer supported at the moment
+        # pending https://github.com/conda-forge/admin-requests/pull/1687
+        - cudnn <9.11               # [cuda_compiler_version != "None"]
       imports:
         - torch
         - torch._C