forked from huawei-noah/vega
-
Notifications
You must be signed in to change notification settings - Fork 0
/
install_dependencies.sh
26 lines (26 loc) · 1.05 KB
/
install_dependencies.sh
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
pip3 install --user albumentations==0.4.2
pip3 install --user torchnet==0.0.4
pip3 install --user Cython==0.29.19
pip3 install --user addict==2.2.1
pip3 install --user opencv-python-headless==4.2.0.34
pip3 install --user distributed==2.18.0
pip3 install --user enum34==1.1.6
pip3 install --user pandas==0.25.2
pip3 install --user scikit-learn==0.21.3
pip3 install --user tensorboardX==1.9
pip3 install --user terminaltables==3.1.0
pip3 install --user timm==0.1.18
pip3 install --user pareto==1.1.1.post3
pip3 install --user py-dag==3.0.1
pip3 install --user thop==0.0.31.post2004101309
pip3 install --user torch==1.2.0
pip3 install --user torchvision==0.4.0
pip3 install --user pytest-runner==5.2
pip3 install --user mmcv==0.2.14
# sudo apt-get install openmpi-bin openmpi-doc libopenmpi-dev
pip3 install --user cffi==1.14.0
HOROVOD_GPU_OPERATIONS=NCCL pip3 install --user --no-cache-dir horovod==0.19.3
pip3 install --user "tensorflow-gpu>=1.14.0,<2.0"
pip3 install --user intervaltree==3.0.2
pip3 install --user more-itertools==8.2.0
pip3 install --user imagecorruptions==1.0.0