diff --git a/conda/pytorch-cuda/conda_build_config.yaml b/conda/pytorch-cuda/conda_build_config.yaml index 93e86f310..b3c3453b5 100644 --- a/conda/pytorch-cuda/conda_build_config.yaml +++ b/conda/pytorch-cuda/conda_build_config.yaml @@ -2,6 +2,7 @@ version: - 11.7 - 11.8 - 12.1 + - 12.4 target_platform: - win-64 - linux-64 diff --git a/conda/pytorch-cuda/meta.yaml b/conda/pytorch-cuda/meta.yaml index 13284b580..763a0c182 100644 --- a/conda/pytorch-cuda/meta.yaml +++ b/conda/pytorch-cuda/meta.yaml @@ -11,7 +11,7 @@ # deployment: # https://conda.anaconda.org/pytorch/noarch/ # https://conda.anaconda.org/pytorch/noarch/repodata.json -{% set build = 5 %} +{% set build = 6 %} {% set cuda_constraints=">=11.7,<11.8" %} {% set libcufft_constraints=">=10.7.2.50,<10.9.0.58" %} {% set libcublas_constraints=">=11.10.1.25,<11.11.3.6" %} @@ -36,6 +36,15 @@ {% set libnpp_constraints=">=12.0.2.50,<12.1.0.40" %} {% set libnvjpeg_constraints=">=12.1.0.39,<12.2.0.2" %} {% set libnvjitlink_constraints=">=12.1.105,<12.2.0" %} +{% elif version == '12.4' %} +{% set cuda_constraints=">=12.4,<12.5" %} +{% set libcufft_constraints=">=11.2.0.44,<11.2.1.3" %} +{% set libcublas_constraints=">=12.4.2.65,<12.4.5.8" %} +{% set libcusolver_constraints=">=11.6.0.99,<11.6.1.9" %} +{% set libcusparse_constraints=">=12.3.0.142,<12.3.1.170" %} +{% set libnpp_constraints=">=12.2.5.2,<12.2.5.30" %} +{% set libnvjpeg_constraints=">=12.3.1.89,<12.3.1.117" %} +{% set libnvjitlink_constraints=">=12.4.99,<12.4.127" %} {% endif %} package: