Skip to content

Commit 8b34ae5

Browse files
authored
add pytorch-cuda constraints for CUDA 12.1 (#1385)
1 parent 6ce166e commit 8b34ae5

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

conda/pytorch-cuda/meta.yaml

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,14 @@
2727
{% set libcusparse_constraints=">=11.7.5.86,<12.0.0.76" %}
2828
{% set libnpp_constraints=">=11.8.0.86,<12.0.0.30" %}
2929
{% set libnvjpeg_constraints=">=11.9.0.86,<12.0.0.28" %}
30+
{% elif version == '12.1' %}
31+
{% set cuda_constraints=">=12.1,<12.2" %}
32+
{% set libcufft_constraints=">=11.0.2.4" %}
33+
{% set libcublas_constraints=">=12.1.0.26" %}
34+
{% set libcusolver_constraints=">=11.4.4.55" %}
35+
{% set libcusparse_constraints=">=12.0.2.55" %}
36+
{% set libnpp_constraints=">=12.0.2.50" %}
37+
{% set libnvjpeg_constraints=">=12.1.0.39" %}
3038
{% endif %}
3139

3240
package:

0 commit comments

Comments
 (0)