@@ -49,35 +49,31 @@ cd ${DEPS_DIR}
49
49
# PREFIX=${DEPS_DIR}/install
50
50
# mkdir -p ${PREFIX}
51
51
52
- # DLPACK_PREFIX=${PREFIX}/dlpack
53
- # TF_PREFIX=${PREFIX}/libtensorflow
54
- # TORCH_PREFIX=${PREFIX}/libtorch
55
- # ORT_PREFIX=${PREFIX}/onnxruntime
52
+ DLPACK=dlpack
53
+ LIBTENSORFLOW=libtensorflow
54
+ LIBTORCH=libtorch
55
+ MKL=mkl
56
+ ONNXRUNTIME=onnxruntime
56
57
57
58
# ####################################################################################### DLPACK
58
59
59
- [[ $FORCE == 1 ]] && rm -rf dlpack
60
+ [[ $FORCE == 1 ]] && rm -rf $DLPACK
60
61
61
- if [[ ! -d dlpack ]]; then
62
+ if [[ ! -d $DLPACK ]]; then
62
63
echo " Cloning dlpack ..."
63
- git clone --depth 1 https://github.com/dmlc/dlpack.git
64
+ git clone --depth 1 https://github.com/dmlc/dlpack.git $DLPACK
64
65
echo " Done."
65
66
else
66
67
echo " dlpack is in place."
67
68
fi
68
69
69
- # if [[ ! -d dlpack/include ]]; then
70
- # mkdir -p dlpack
71
- # ln -sf ${DEPS_DIR}/dlpack/include ${DLPACK_PREFIX}/include
72
- # fi
73
-
74
70
# ################################################################################### TENSORFLOW
75
71
76
72
TF_VERSION=" 1.14.0"
77
73
78
- [[ $FORCE == 1 ]] && rm -rf tensorflow
74
+ [[ $FORCE == 1 ]] && rm -rf $LIBTENSORFLOW
79
75
80
- if [[ ! -d tensorflow ]]; then
76
+ if [[ ! -d $LIBTENSORFLOW ]]; then
81
77
echo " Installing TensorFlow ..."
82
78
83
79
if [[ $OS == linux ]]; then
@@ -112,10 +108,10 @@ if [[ ! -d tensorflow ]]; then
112
108
113
109
[[ ! -f $LIBTF_ARCHIVE || $FORCE == 1 ]] && wget --quiet $LIBTF_URL_BASE /$LIBTF_ARCHIVE
114
110
115
- rm -rf tensorflow .x
116
- mkdir tensorflow .x
117
- tar xf $LIBTF_ARCHIVE --no-same-owner --strip-components=1 -C tensorflow .x
118
- mv tensorflow .x tensorflow
111
+ rm -rf $LIBTENSORFLOW .x
112
+ mkdir $LIBTENSORFLOW .x
113
+ tar xf $LIBTF_ARCHIVE --no-same-owner --strip-components=1 -C $LIBTENSORFLOW .x
114
+ mv $LIBTENSORFLOW .x $LIBTENSORFLOW
119
115
120
116
echo " Done."
121
117
else
126
122
127
123
PT_VERSION=" 1.2.0"
128
124
129
- [[ $FORCE == 1 ]] && rm -rf libtorch
125
+ [[ $FORCE == 1 ]] && rm -rf $LIBTORCH
130
126
131
- if [[ ! -d libtorch ]]; then
127
+ if [[ ! -d $LIBTORCH ]]; then
132
128
echo " Installing libtorch ..."
133
129
134
130
if [[ $OS == linux ]]; then
@@ -157,12 +153,12 @@ if [[ ! -d libtorch ]]; then
157
153
158
154
[[ ! -f $LIBTORCH_ARCHIVE || $FORCE == 1 ]] && wget -q $LIBTORCH_URL
159
155
160
- rm -rf libtorch .x
161
- mkdir libtorch .x
156
+ rm -rf $LIBTORCH .x
157
+ mkdir $LIBTORCH .x
162
158
163
- tar xf $LIBTORCH_ARCHIVE --no-same-owner -C libtorch .x
164
- mv libtorch .x/libtorch libtorch
165
- rmdir libtorch .x
159
+ tar xf $LIBTORCH_ARCHIVE --no-same-owner -C $LIBTORCH .x
160
+ mv $LIBTORCH .x/libtorch $LIBTORCH
161
+ rmdir $LIBTORCH .x
166
162
167
163
echo " Done."
168
164
else
@@ -181,10 +177,10 @@ if [[ ! -d mkl ]]; then
181
177
MKL_ARCHIVE=mklml_${MKL_OS} _${MKL_BUNDLE_VER} .tgz
182
178
[[ ! -e ${MKL_ARCHIVE} ]] && wget -q https://github.com/intel/mkl-dnn/releases/download/v${MKL_VERSION} /${MKL_ARCHIVE}
183
179
184
- rm -rf mkl .x
185
- mkdir mkl .x
186
- tar xzf ${MKL_ARCHIVE} --no-same-owner --strip-components=1 -C mkl .x
187
- mv mkl .x mkl
180
+ rm -rf $MKL .x
181
+ mkdir $MKL .x
182
+ tar xzf ${MKL_ARCHIVE} --no-same-owner --strip-components=1 -C $MKL .x
183
+ mv $MKL .x $MKL
188
184
189
185
echo " Done."
190
186
fi
196
192
197
193
ORT_VERSION=" 0.5.0"
198
194
199
- [[ $FORCE == 1 ]] && rm -rf onnxruntime
195
+ [[ $FORCE == 1 ]] && rm -rf $ONNXRUNTIME
200
196
201
- if [[ ! -d onnxruntime ]]; then
197
+ if [[ ! -d $ONNXRUNTIME ]]; then
202
198
echo " Installing ONNXRuntime ..."
203
199
204
200
if [[ $OS == linux ]]; then
@@ -227,15 +223,12 @@ if [[ ! -d onnxruntime ]]; then
227
223
228
224
ORT_ARCHIVE=onnxruntime-${ORT_OS} -${ORT_ARCH}${ORT_BUILD} -${ORT_VERSION} .tgz
229
225
230
- if [[ ! -e ${ORT_ARCHIVE} ]]; then
231
- wget -q $ORT_URL_BASE /${ORT_ARCHIVE}
232
- echo " Done."
233
- fi
226
+ [[ ! -e ${ORT_ARCHIVE} ]] && wget -q $ORT_URL_BASE /${ORT_ARCHIVE}
234
227
235
- rm -rf onnxruntime .x
236
- mkdir onnxruntime .x
237
- tar xzf ${ORT_ARCHIVE} --no-same-owner --strip-components=1 -C onnxruntime .x
238
- mv onnxruntime .x onnxruntime
228
+ rm -rf $ONNXRUNTIME .x
229
+ mkdir $ONNXRUNTIME .x
230
+ tar xzf ${ORT_ARCHIVE} --no-same-owner --strip-components=1 -C $ONNXRUNTIME .x
231
+ mv $ONNXRUNTIME .x $ONNXRUNTIME
239
232
240
233
echo " Done."
241
234
else
256
249
# echo "Binaries in place."
257
250
# fi
258
251
259
- echo " Done."
252
+ # echo "Done."
0 commit comments