Skip to content

Commit efec1f2

Browse files
committed
update meta.yaml template
1 parent 848d22b commit efec1f2

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

conda-recipe/meta.yaml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,13 @@ requirements:
2828
- {{ dep.lower() }}
2929
{% endfor %}
3030
run_constrained:
31+
- cudatoolkit>=10.1
3132
{% for dep in setup_py_data['extras_require']['pytorch'] %}
33+
{% if dep.startswith('torch>=') %} # pip: torch -> conda: pytorch
34+
- py{{ dep.lower() }}
35+
{% else %}
3236
- {{ dep.lower() }}
37+
{% endif %}
3338
{% endfor %}
3439
{% for dep in setup_py_data['extras_require']['onnx'] %}
3540
- {{ dep.lower() }}

0 commit comments

Comments
 (0)