forked from twosixlabs/armory
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathhost-requirements.txt
35 lines (32 loc) · 968 Bytes
/
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
coloredlogs
docker #To prevent import errors
jsonschema
requests
tensorflow-datasets==3.2.0
jupyterlab==1.2.6
boto3==1.11.13
adversarial-robustness-toolbox==1.4.2
Pillow==7.1.2
pydub==0.24.1
apache-beam==2.22.0
dill==0.3.1.1
pytest==5.3.5
ffmpeg==4.2.3
pymongo==3.1.0
### Begin TF1 required libraries
tensorflow-gpu==1.15.0
# TF1 object detection API
# https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/tf1.md#python-package-installation
# Commit hash 79354e14a4b41ff9019f4a5ebd12cfa498917182
### End TF1 required libraries
### Begin Pytorch required libraries
pytorch==1.6.0
torchvision==0.7.0
cudatoolkit==10.1
### End Pytorch required libraries
### Begin TF2 required libraries
tensorflow-gpu==2.2.0
# TF2 object detection API
# https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/tf2.md#python-package-installation
# Commit hash 79354e14a4b41ff9019f4a5ebd12cfa498917182
### End TF2 required libraries