Skip to content

Commit

Permalink
Update pytorch-cuda for 12.4 conda build. (#1792)
Browse files Browse the repository at this point in the history
  • Loading branch information
nWEIdia authored Apr 20, 2024
1 parent a79e1ce commit 94d6e5c
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 1 deletion.
1 change: 1 addition & 0 deletions conda/pytorch-cuda/conda_build_config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ version:
- 11.7
- 11.8
- 12.1
- 12.4
target_platform:
- win-64
- linux-64
11 changes: 10 additions & 1 deletion conda/pytorch-cuda/meta.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -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" %}
Expand All @@ -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:
Expand Down

0 comments on commit 94d6e5c

Please sign in to comment.