forked from twosixlabs/armory
-
Notifications
You must be signed in to change notification settings - Fork 0
/
host-requirements.txt
75 lines (70 loc) · 2.21 KB
/
host-requirements.txt
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
73
74
75
#
# This file is only to be used if the local environment requires that you
# run armory in "--no-docker" mode. Using no-docker mode adds considerable
# complexity to the installation and operation of armory, so we encourage the
# standard containerized version for most uses.
# Because Armory supports PyTorch, TensorFlow1, and TensorFlow2, we've
# segmented the per-platform requirements below which you should uncomment
# depending on which you use. TensorFlow1 support by Armory is deprecated and
# will be removed in a future release.
coloredlogs
docker
jsonschema
requests
tensorflow-datasets==3.2.0
jupyterlab==3.0.9
boto3==1.17.20
Pillow==8.2.0
adversarial-robustness-toolbox==1.6.2
pydub==0.24.1
apache-beam==2.22.0
dill==0.3.1.1
pytest==6.2.2
ffmpeg-python==0.2.0
opencv-python==4.5.1.48
tensorflow-datasets==3.2.0
jupyterlab==3.0.9
boto3==1.17.20
Pillow==8.1.1
pydub==0.24.1
apache-beam==2.22.0
dill==0.3.1.1
pytest==6.2.2
pandas==1.2.4
ffmpeg-python==0.2.0
###
### Begin Pytorch required libraries
torch==1.7.0
torchvision==0.8.1
torchaudio==0.6.0
cudatoolkit==11.0
hydra-core~=1.0.6
python-levenshtein~=0.12.2
librosa
sox
matplotlib
# The deepspeech asr processing requires warpctc-pytorch which is highly environment
# dependent. If you are not useing that model or related, you don't need to import
# waprctc-pytorch
# warpctc-pytorch==0.2.1+torch16.cuda102
###
### End Pytorch required libraries
###
### Begin TF2 required libraries
#tensorflow-gpu==2.4.1
# The TF2 object detection API requires separate installation, which
# is not simply handled by pip. Instructions can be found at
# https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/tf2.md#python-package-installation
# Commit hash 79354e14a4b41ff9019f4a5ebd12cfa498917182
### End TF2 required libraries
###
### Begin TF1 required libraries
###
# tensorflow-gpu==1.15.0
# The TF1 object detection API requires separate installation, which
# is not simply handled by pip. Instructions can be found at
# https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/tf1.md#python-package-installation
# Commit hash 79354e14a4b41ff9019f4a5ebd12cfa498917182
###
### End TF1 required libraries
###