Add pytorch-cuda constraints for CUDA 12.1

conda/pytorch-cuda/meta.yaml

@@ -27,6 +27,14 @@
 {% set libcusparse_constraints=">=11.7.5.86,<12.0.0.76" %}
 {% set libnpp_constraints=">=11.8.0.86,<12.0.0.30" %}
 {% set libnvjpeg_constraints=">=11.9.0.86,<12.0.0.28" %}
+{% elif version == '12.1' %}
+{% set cuda_constraints=">=12.1,<12.2" %}
+{% set libcufft_constraints=">=11.0.2.4" %}
+{% set libcublas_constraints=">=12.1.0.26" %}
+{% set libcusolver_constraints=">=11.4.4.55" %}
+{% set libcusparse_constraints=">=12.0.2.55" %}
+{% set libnpp_constraints=">=12.0.2.50" %}
+{% set libnvjpeg_constraints=">=12.1.0.39" %}
 {% endif %}
 
 package: