-
Notifications
You must be signed in to change notification settings - Fork 11
/
meta.yaml
72 lines (65 loc) · 1.83 KB
/
meta.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
# changed the recipe to be client, server at the same time
# add pytorch (or other NN libraries, see bioimageio.core) to enable server
# capabilities.
# Add git, inferno, pytorch!
{% set setup_py_data = load_setup_py_data() %}
package:
name: tiktorch
version: {{ setup_py_data['version'] }}
source:
path: ../
build:
noarch: python
number: 0
script: python -m pip install --no-deps --ignore-installed .
entry_points:
- tiktorch-server = tiktorch.server.base:main
requirements:
host:
- pip
- python >=3.7
run:
- python >=3.7
- pickle5 # [py37]
{% for dep in setup_py_data['install_requires'] %}
- {{ dep.lower() }}
{% endfor %}
run_constrained:
- cudatoolkit >=10.2
{% for dep in setup_py_data['extras_require']['server-pytorch'] %}
- {{ dep.lower() }}
{% endfor %}
{% for dep in setup_py_data['extras_require']['server-tensorflow'] %}
- {{ dep.lower() }}
{% endfor %}
about:
home: https://github.com/ilastik/tiktorch
license: MIT
license_family: MIT
license_file: LICENSE
summary: 'Neural network server'
doc_url: https://github.com/ilastik/tiktorch
dev_url: https://github.com/ilastik/tiktorch
test:
requires:
{% for dep in setup_py_data['extras_require']['server-pytorch'] %}
- {{ dep.lower() }}
{% endfor %}
{% for dep in setup_py_data['extras_require']['server-tensorflow'] %}
- {{ dep.lower() }}
{% endfor %}
# this is still necessary, torchvision doesn't work properly with cpuonly mutex
- torchvision=*=*cpu
- cpuonly
imports:
# client
- tiktorch
- tiktorch.proto.inference_pb2
- tiktorch.proto.inference_pb2_grpc
- tiktorch.proto.data_store_pb2
- tiktorch.proto.data_store_pb2_grpc
- tiktorch.types
- tiktorch.converters
# server
- tiktorch.server.base
- tiktorch.server.session