-
Notifications
You must be signed in to change notification settings - Fork 2
/
install_env.sh
112 lines (93 loc) · 2.6 KB
/
install_env.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
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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
#!/bin/bash
echo "****************** Installing pytorch with cuda10 ******************"
conda install -y pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch
echo ""
echo ""
echo "****************** Installing pandas ******************"
pip install pandas
echo ""
echo ""
echo "****************** Installing opencv ******************"
pip install opencv-python
echo ""
echo ""
echo "****************** Installing cython ******************"
pip install cython==0.29.21
echo ""
echo ""
echo "****************** Installing skimage ******************"
pip install scikit-image
echo ""
echo ""
echo "****************** Installing pillow ******************"
pip install 'pillow<7.0.0'
echo ""
echo ""
echo "****************** Installing scipy ******************"
pip install scipy
echo ""
echo ""
echo "****************** Installing shapely ******************"
pip install shapely
echo ""
echo ""
echo "****************** Installing easydict ******************"
pip install easydict
echo ""
echo ""
echo "****************** Installing jpeg4py python wrapper ******************"
pip install jpeg4py
echo ""
echo ""
echo '****************** Intalling mpi4py ******************'
conda install mpi4py
echo ""
echo ""
echo '****************** Intalling ray and hyperopt ******************'
pip uninstall --yes grpcio
pip install --upgrade setuptools
pip install --no-cache-dir grpcio>=1.28.1
conda install -c conda-forge grpcio
pip install ray==0.8.7
pip install ray[tune]
pip install hyperopt
echo ""
echo ""
echo "****************** Installing numba ******************"
pip install numba
echo ""
echo ""
echo "****************** Installing EfficientNet ******************"
cd lib/models/EfficientNet-PyTorch
pip install -e .
cd ../../..
echo ""
echo ""
echo "****************** Installing thop tool for FLOPs and Params computing ******************"
pip install --upgrade git+https://github.com/Lyken17/pytorch-OpCounter.git
echo ""
echo ""
echo "****************** Installing GOT-10K related packages ******************"
cd toolkit
pip install -r requirements.txt
conda install matplotlib==3.0.2
cd ..
echo ""
echo ""
echo "****************** Installing Cream ******************"
pip install yacs
conda install tensorboard
pip install timm==0.1.20
pip install git+https://github.com/sovrasov/flops-counter.pytorch.git
pip install git+https://github.com/Tramac/torchscope.git
cd supernet_backbone
python setup.py
cd ..
echo ""
echo ""
echo "****************** Installing tensorboardX, colorama ******************"
pip install tensorboardX
pip install colorama
echo ""
echo ""
echo "****************** Installation complete! ******************"