forked from MCG-NJU/MixFormer
-
Notifications
You must be signed in to change notification settings - Fork 0
/
install_pytorch17.sh
103 lines (82 loc) · 2.46 KB
/
install_pytorch17.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
echo "****************** Installing pytorch ******************"
conda install -y pytorch==1.7.0 torchvision==0.8.1 cudatoolkit=10.2 -c pytorch
echo ""
echo ""
echo "****************** Installing yaml ******************"
pip install PyYAML
echo ""
echo ""
echo "****************** Installing easydict ******************"
pip install easydict
echo ""
echo ""
echo "****************** Installing cython ******************"
pip install cython
echo ""
echo ""
echo "****************** Installing opencv-python ******************"
pip install opencv-python
echo ""
echo ""
echo "****************** Installing pandas ******************"
pip install pandas
echo ""
echo ""
echo "****************** Installing tqdm ******************"
conda install -y tqdm
echo ""
echo ""
echo "****************** Installing coco toolkit ******************"
pip install pycocotools
echo ""
echo ""
echo "****************** Installing jpeg4py python wrapper ******************"
apt-get install libturbojpeg
pip install jpeg4py
echo ""
echo ""
echo "****************** Installing tensorboard ******************"
pip install tb-nightly
echo ""
echo ""
echo "****************** Installing tikzplotlib ******************"
pip install tikzplotlib
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 colorama ******************"
pip install colorama
echo ""
echo ""
echo "****************** Installing lmdb ******************"
pip install lmdb
echo ""
echo ""
echo "****************** Installing scipy ******************"
pip install scipy
echo ""
echo ""
echo "****************** Installing visdom ******************"
pip install visdom
echo ""
echo ""
echo "****************** Installing vot-toolkit python ******************"
pip install git+https://github.com/votchallenge/vot-toolkit-python
echo ""
echo ""
echo "****************** Installing onnx and onnxruntime-gpu ******************"
pip install onnx onnxruntime-gpu==1.6.0
echo ""
echo ""
echo "****************** Installing timm ******************"
pip install timm==0.3.2
echo "****************** Installing yacs/einops/thop ******************"
pip install yacs
pip install einops
pip install thop
echo "****************** Install ninja-build for Precise ROI pooling ******************"
apt-get install ninja-build
echo "****************** Installation complete! ******************"