From 4838d6e628eb53a761b42a7a60496416bbdba6c5 Mon Sep 17 00:00:00 2001 From: pbialecki Date: Fri, 7 Apr 2023 02:46:09 -0700 Subject: [PATCH] add pytorch-cuda constraints for CUDA 12.1 --- conda/pytorch-cuda/meta.yaml | 8 ++++++++ 1 file changed, 8 insertions(+) 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: