diff --git a/conda/pytorch-cuda/meta.yaml b/conda/pytorch-cuda/meta.yaml index 4c3ff4d8a..778bf705c 100644 --- a/conda/pytorch-cuda/meta.yaml +++ b/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: