/
install.sh
67 lines (49 loc) · 1.91 KB
/
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
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
#! /bin/bash
# CK installation script for PyTorch package
#
# Developer(s):
# * Grigori Fursin, dividiti/cTuning foundation
#
# PACKAGE_DIR
# INSTALL_DIR
# TF_PYTHON_URL
# This is where pip will install the modules.
# It has its own funny structure we don't control :
#
EXTRA_PYTHON_SITE=${INSTALL_DIR}/python_deps_site
SHORT_PYTHON_VERSION=`${CK_ENV_COMPILER_PYTHON_FILE} -c 'import sys;print(sys.version[:3])'`
export PACKAGE_LIB_DIR="${EXTRA_PYTHON_SITE}/lib/python${SHORT_PYTHON_VERSION}/site-packages"
export PYTHONPATH=$PACKAGE_LIB_DIR:$PYTHONPATH
######################################################################################
echo ""
echo "Removing '${EXTRA_PYTHON_SITE}' ..."
rm -rf ${EXTRA_PYTHON_SITE}
######################################################################################
# Check if has --system option
${CK_ENV_COMPILER_PYTHON_FILE} -m pip install --help > tmp-pip-help.tmp
if grep -q "\-\-system" tmp-pip-help.tmp ; then
SYS=" --system"
fi
rm -f tmp-pip-help.tmp
######################################################################################
echo "Downloading and installing deps ..."
echo ""
${CK_ENV_COMPILER_PYTHON_FILE} -m pip install --ignore-installed requests --prefix=${EXTRA_PYTHON_SITE} ${SYS}
if [ "${?}" != "0" ] ; then
echo "Error: installation failed!"
exit 1
fi
######################################################################################
MAJOR_PYTHON_VERSION=`${CK_ENV_COMPILER_PYTHON_FILE} -c 'import sys;print(sys.version[0])'`
######################################################################################
echo "Cloning PyTorch Vision ..."
echo ""
cd ${INSTALL_DIR}
git clone --recursive ${PACKAGE_URL}
cd vision
#mkdir -p ${EXTRA_PYTHON_SITE}
#mkdir -p ${PACKAGE_LIB_DIR}
${CK_ENV_COMPILER_PYTHON_FILE} setup.py install --prefix=${EXTRA_PYTHON_SITE}
cp -rf build/lib/* ${PACKAGE_LIB_DIR}
ln -s $PACKAGE_LIB_DIR ${INSTALL_DIR}/lib
exit 0