/
update-runtime.sh
executable file
·40 lines (34 loc) · 1.19 KB
/
update-runtime.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
#!/bin/bash
ignore_hordelib=false
# Parse command line arguments
while [[ $# -gt 0 ]]
do
key="$1"
case $key in
--hordelib)
hordelib=true
shift # past argument
;;
--scribe)
scribe=true
shift
;;
*) # unknown option
echo "Unknown option: $key"
exit 1
;;
esac
shift # past argument or value
done
CONDA_ENVIRONMENT_FILE=environment.yaml
wget -qO- https://github.com/mamba-org/micromamba-releases/releases/latest/download/micromamba-linux-64.tar.bz2 | tar -xvj bin/micromamba
if [ ! -f "conda/envs/linux/bin/python" ]; then
bin/micromamba create --no-shortcuts -r conda -n linux -f ${CONDA_ENVIRONMENT_FILE} -y
fi
bin/micromamba create --no-shortcuts -r conda -n linux -f ${CONDA_ENVIRONMENT_FILE} -y
if [ "$hordelib" = true ]; then
bin/micromamba run -r conda -n linux python -s -m pip uninstall -y hordelib horde_engine horde_sdk horde_model_reference
bin/micromamba run -r conda -n linux python -s -m pip install horde_engine horde_model_reference --extra-index-url https://download.pytorch.org/whl/cu121
else
bin/micromamba run -r conda -n linux python -s -m pip install -r requirements.txt -U --extra-index-url https://download.pytorch.org/whl/cu121
fi