This repository has been archived by the owner on Jun 28, 2024. It is now read-only.
forked from emvollmer/tbbrdet_api
-
Notifications
You must be signed in to change notification settings - Fork 0
/
install_TBBRDet.sh
111 lines (87 loc) · 3.38 KB
/
install_TBBRDet.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
#!/bin/bash
check_correct_version () {
local what=$1
local required_ver=$2
local current_ver=$3
if [[ "$required_ver" == "$current_ver" ]]; then
echo "$what version requirement $required_ver satisfied."
else
echo "Provided $what version $current_ver is not compatible with required ${required_ver}).
Please install $what $required_ver "; sleep 5
exit 1
fi
}
# Check Python version
current_python=$(python --version 2>&1 | cut -d ' ' -f 2 | cut -d '.' -f 1,2)
required_python="3.6"
check_correct_version "Python" $required_python "$current_python"
# Check CUDA version
current_cuda=$(nvcc --version | grep 'release' | awk '{print $5}' | cut -d ',' -f 1)
required_cuda="11.6"
check_correct_version "CUDA" $required_cuda "$current_cuda"
# Make sure all applications are there for next steps
echo "Installing necessary applications for virtual environment setup"
yes | apt-get install python3.6-dev
if message=$(yes | apt-get install python3.6-virtualenv 2>&1); then
echo "Python3.6-virtualenv successfully installed."
else
echo "Python3.6-virtualenv installation unsuccessful:'$message'"
yes | apt install python3.6-venv
fi
# Switch to api directory if not already there
if ! [[ "$PWD" == *\tbbrdet_api ]] && [ -d "tbbrdet_api" ] ; then
echo "Not yet in tbbrdet_api. Changing into directory to continue working..."
cd tbbrdet_api
fi
# Check for activated venv, create one if there is none, otherwise activate it
if [[ "$VIRTUAL_ENV" != "" ]]; then
echo "Currently in an activated virtual environment. Setup may proceed."
else
echo "Currently not in an activated virtual environment. Must have one up to proceed with package installation..."
venv_name="venv"
venv_act="$venv_name"/bin/activate
if test -f $venv_act; then
echo "Virtual environment at '$venv_name' already exists. Activating..."
else
if message=$(python3.6 -m venv "$venv_name" 2>&1); then
echo "Virtual environment successfully created. Activating..."
else
echo "Python3.6 venv created unsuccessful. Stopping..."; sleep 10
exit 1
fi
fi
source $venv_act
fi
install_and_check_package() {
local package_name=${1#* } # Remove the command part from the package name
echo "Installing $package_name..."
"$1"
# Check the exit code of the previous command
if [ $? -ne 0 ]; then
echo "Installation of $package_name failed."; sleep 10
# exit 1 # Exit the script with a non-zero status to indicate failure
else
echo "$package_name installed successfully."
fi
}
# Ensure packages are installed in the correct order
# Make sure pip is up to date
pip3 install pip==21.3.1 setuptools==59.6.0 wheel==0.37.1
# Install correct torch version
pip3 install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html --no-cache-dir
# Install openmim using pip
pip3 install -U openmim
# Use mim to install mmcv
mim install mmcv-full==1.5
# Install correct mmdetection version
pip3 install mmdet==2.21.0
# Install everything else
pip3 install future tensorboard mlflow joblib tqdm
echo "Packages installed."
echo "Installing TBBRDet repository as editable..."
pip3 install -e ./TBBRDet
echo "Installing tbbrdet_api repository as editable..."
pip3 install -e .
echo "==============================="
echo "Installation process complete."
echo "==============================="