File tree Expand file tree Collapse file tree 4 files changed +7
-9
lines changed Expand file tree Collapse file tree 4 files changed +7
-9
lines changed Original file line number Diff line number Diff line change @@ -97,7 +97,7 @@ ADD_DEFINITIONS(-DREDISMODULE_EXPERIMENTAL_API)
97
97
IF (BUILD_TF)
98
98
FIND_LIBRARY (TF_LIBRARIES NAMES tensorflow
99
99
PATHS ${depsAbs} /libtensorflow/lib)
100
- MESSAGE (STATUS "Found TensorFlow Libraries: \" ${TF_LIBRARIES} \" )" )
100
+ MESSAGE (STATUS "Found TensorFlow Libraries: \" ${TF_LIBRARIES} \" )" )
101
101
IF (NOT TF_LIBRARIES)
102
102
MESSAGE (WARNING "Could not find TensorFlow in ${depsAbs} /libtensorflow/lib. Trying find_package method" )
103
103
FIND_PACKAGE (TensorFlow REQUIRED)
@@ -113,11 +113,11 @@ ENDIF()
113
113
#----------------------------------------------------------------------------------------------
114
114
115
115
IF (BUILD_TFLITE)
116
- FIND_LIBRARY (TFLITE_LIBRARIES_1 NAMES tensorflow-lite
117
- PATHS ${depsAbs} /libtensorflow-lite/lib)
118
- FIND_LIBRARY (TFLITE_LIBRARIES_2 NAMES benchmark-lib.a
119
- PATHS ${depsAbs} /libtensorflow-lite/lib)
120
- SET (TFLITE_LIBRARIES ${TFLITE_LIBRARIES_1} ${TFLITE_LIBRARIES_2} )
116
+ FIND_LIBRARY (TFLITE_LIBRARIES_1 NAMES tensorflow-lite
117
+ PATHS ${depsAbs} /libtensorflow-lite/lib)
118
+ FIND_LIBRARY (TFLITE_LIBRARIES_2 NAMES benchmark-lib.a
119
+ PATHS ${depsAbs} /libtensorflow-lite/lib)
120
+ SET (TFLITE_LIBRARIES ${TFLITE_LIBRARIES_1} ${TFLITE_LIBRARIES_2} )
121
121
MESSAGE (STATUS "Found TensorFlow Lite Libraries: \" ${TFLITE_LIBRARIES} \" )" )
122
122
IF (NOT TFLITE_LIBRARIES)
123
123
MESSAGE (FATAL_ERROR "Could not find TensorFlow Lite" )
Original file line number Diff line number Diff line change @@ -170,7 +170,6 @@ if [[ $WITH_TFLITE != 0 ]]; then
170
170
LIBTFLITE_ARCHIVE=libtensorflowlite-${TFLITE_OS} -${TFLITE_ARCH} -${TFLITE_VERSION} .tar.gz
171
171
172
172
[[ ! -f $LIBTFLITE_ARCHIVE || $FORCE == 1 ]] && wget $LIBTF_URL_BASE /$LIBTFLITE_ARCHIVE
173
- echo $LIBTF_URL_BASE /$LIBTFLITE_ARCHIVE
174
173
175
174
rm -rf $LIBTFLITE .x
176
175
mkdir $LIBTFLITE .x
Original file line number Diff line number Diff line change @@ -16,6 +16,7 @@ elif [[ "$1" == "gpu" ]]; then
16
16
else
17
17
GPU=${GPU:- 0}
18
18
fi
19
+
19
20
OS=$( python3 $ROOT /opt/readies/bin/platform --os)
20
21
ARCH=$( python3 $ROOT /opt/readies/bin/platform --arch)
21
22
65
66
LIBTORCH_ZIP=libtorch-${PT_BUILD} -${PT_OS} -${PT_ARCH} -${PT_VERSION} .zip
66
67
if [ ! -f $LIBTORCH_ARCHIVE ]; then
67
68
echo " Downloading libtorch ${PT_VERSION} ${PT_BUILD} "
68
- echo $LIBTORCH_URL
69
69
wget -q -O $LIBTORCH_ZIP $LIBTORCH_URL
70
70
fi
71
71
Original file line number Diff line number Diff line change 1
1
numpy
2
- scipy
3
2
scikit-image
4
3
redis-py-cluster
You can’t perform that action at this time.
0 commit comments