install.sh

#!/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.7
conda activate $env_name

echo "***** install numpy pytorch opencv *****"
# numpy
conda install -y numpy
# pytorch
# pytorch with cuda101 is tested
conda install -y pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -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
cd pysot/models/head/dcn/ && python setup.py build_ext --inplace