-
Notifications
You must be signed in to change notification settings - Fork 16
/
install.sh
42 lines (32 loc) · 894 Bytes
/
install.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
#!/bin/bash
if [ $# -lt 2 ]; then
echo "ARGS ERROR!"
echo " bash install.sh /path/to/your/conda env_name"
exit 1
fi
set -e
conda_path=$1
env_name=$2
source $conda_path/etc/profile.d/conda.sh
echo "****** create environment " $env_name "*****"
# create environment
conda create -y --name $env_name python=3.6
conda activate $env_name
echo "***** install numpy pytorch opencv *****"
# numpy
conda install -y numpy
# pytorch
# pytorch with cuda80/cuda90 is tested
conda install -y pytorch=1.1.0 torchvision cuda90 -c pytorch
# opencv
pip install opencv-python
# tensorboardX
echo "***** install other libs *****"
pip install tensorboardX
# libs
pip install pyyaml yacs tqdm colorama matplotlib cython
echo "***** build extensions *****"
python setup.py build_ext --inplace
echo "***** build extensions about DCN *****"
cd siamreppoints
python ./setup.py build_ext --inplace