Selaa lähdekoodia

Add dependencies to -update scripts

Lutz Roeder 7 vuotta sitten
vanhempi
sitoutus
0a3abff8c3
3 muutettua tiedostoa jossa 8 lisäystä ja 2 poistoa
  1. 1 0
      tools/keras-update
  2. 7 0
      tools/tf-update
  3. 0 2
      tools/update

+ 1 - 0
tools/keras-update

@@ -27,6 +27,7 @@ echo "Install Keras"
 export PYTHONUSERBASE=${build}/third_party/pypi/${identifier}
 export PATH=${PATH}:${build}/third_party/pypi/${identifier}/bin
 rm -rf ${PYTHONUSERBASE}
+${pip} install --user tensorflow
 ${pip} install --user ${third_party}/${identifier}
 
 echo "Update '../src/keras-metadata.json'"

+ 7 - 0
tools/tf-update

@@ -3,6 +3,7 @@
 set -e
 
 root=$(cd $(dirname ${0})/..; pwd)
+build=${root}/build
 node_modules=${root}/node_modules
 src=${root}/src
 tools=${root}/tools
@@ -40,6 +41,12 @@ ${node_modules}/protobufjs/bin/pbjs -t static-module -w closure -r tf -o ${src}/
     ${third_party}/${identifier}/tensorflow/core/framework/tensor.proto \
     ${third_party}/${identifier}/tensorflow/core/framework/resource_handle.proto \
 
+export PYTHONUSERBASE=${build}/third_party/pypi/${identifier}
+export PATH=$PATH:${PYTHONUSERBASE}/bin
+echo $PYTHONUSERBASE
+rm -rf ${PYTHONUSERBASE}
+${pip} install --user protobuf
+
 echo "Generate '../src/tf-metadata.json'"
 protoc --proto_path ${third_party}/${identifier} tensorflow/core/framework/attr_value.proto --python_out=${tools}
 protoc --proto_path ${third_party}/${identifier} tensorflow/core/framework/tensor.proto --python_out=${tools}

+ 0 - 2
tools/update

@@ -13,8 +13,6 @@ ${tools}/tf-update
 ${tools}/coreml-update
 ${tools}/caffe-update
 ${tools}/mxnet-update
-
 ${tools}/onnx-update
 ${tools}/tflite-update
-
 ${tools}/caffe2-update