Merge pull request #292 from bioimage-io/fix_pip

Fix pip installation

Author: FynnBe

conda-recipe/meta.yaml

@@ -28,8 +28,13 @@ requirements:
     - {{ dep.lower() }}
     {% endfor %}
   run_constrained:
+    - cudatoolkit>=10.1
     {% for dep in setup_py_data['extras_require']['pytorch'] %}
+      {% if dep.startswith('torch>=') %}  # pip: torch  -> conda: pytorch
+    - py{{ dep.lower() }}
+      {% else %}
     - {{ dep.lower() }}
+      {% endif %}
     {% endfor %}
     {% for dep in setup_py_data['extras_require']['onnx'] %}
     - {{ dep.lower() }}

setup.py

@@ -38,7 +38,7 @@
     extras_require={
         "test": ["pytest", "black", "mypy"],
         "dev": ["pre-commit"],
-        "pytorch": ["pytorch>=1.6", "torchvision", "cudatoolkit>=10.1"],
+        "pytorch": ["torch>=1.6", "torchvision"],
         "tensorflow": ["tensorflow"],
         "onnx": ["onnxruntime"],
     },