diff --git a/conda/pytorch-cuda/meta.yaml b/conda/pytorch-cuda/meta.yaml index ff31b34b0..a5c8cadcf 100644 --- a/conda/pytorch-cuda/meta.yaml +++ b/conda/pytorch-cuda/meta.yaml @@ -29,12 +29,12 @@ {% 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" %} +{% set libcufft_constraints=">=11.0.2.4,<11.0.2.54" %} +{% set libcublas_constraints=">=12.1.0.26,<12.1.3.1" %} +{% set libcusolver_constraints=">=11.4.4.55,<11.4.5.107" %} +{% set libcusparse_constraints=">=12.0.2.55,<12.1.0.106" %} +{% set libnpp_constraints=">=12.0.2.50,<12.1.0.40" %} +{% set libnvjpeg_constraints=">=12.1.0.39,<12.2.0.2" %} {% endif %} package: