Procházet zdrojové kódy

Workaround onnx/onnx#2481

Lutz Roeder před 6 roky
rodič
revize
e481f85311
1 změnil soubory, kde provedl 8 přidání a 5 odebrání
  1. 8 5
      tools/onnx

+ 8 - 5
tools/onnx

@@ -50,18 +50,21 @@ sync() {
 
 install() {
     bold "onnx install"
-    case "$(uname)" in
-        "Linux")
+    case "${OSTYPE}" in
+        linux*)
             [ -n "$(which cmake)" ] || sudo apt install -y cmake
             [ -n "$(which protoc)" ] || sudo apt install -y protobuf-compiler libprotoc-dev
+            protobuf=protobuf
             ;;
-        "Darwin")
+        darwin*)
             brew list cmake > /dev/null 2>&1 || brew install cmake > /dev/null
-            brew list protobuf > /dev/null 2>&1 || brew install protobuf > /dev/null
+            brew list [email protected] > /dev/null 2>&1 || brew install [email protected] > /dev/null
+            export PATH=/usr/local/opt/[email protected]/bin:${PATH}
+            protobuf="protobuf==3.7.1"
             ;;
     esac
     venv
-    ${python} -m pip install --quiet --upgrade protobuf
+    ${python} -m pip install --quiet --upgrade ${protobuf}
     export ONNX_ML=1
     export ONNX_NAMESPACE=onnx
     ${python} -m pip install --quiet ./third_party/onnx