瀏覽代碼

Add pip upgrade to scripts

Lutz Roeder 6 年之前
父節點
當前提交
65ea7dd076
共有 5 個文件被更改,包括 8 次插入8 次删除
  1. 1 1
      tools/coreml
  2. 1 1
      tools/keras
  3. 1 1
      tools/onnx
  4. 4 4
      tools/pytorch
  5. 1 1
      tools/tf

+ 1 - 1
tools/coreml

@@ -41,7 +41,7 @@ install() {
     [ -n "$(python3 -m pip list --format columns --disable-pip-version-check | grep -w virtualenv)" ] || python3 -m pip install --force-reinstall --user --quiet virtualenv
     [ -d "${virtualenv}" ] || python3 -m virtualenv --quiet -p python3 ${virtualenv}
     source ${virtualenv}/bin/activate
-    python3 -m pip install --quiet six numpy protobuf
+    python3 -m pip install --quiet --upgrade six numpy protobuf
     python3 -m pip install --quiet ${third_party}/${identifier}
     deactivate
 }

+ 1 - 1
tools/keras

@@ -42,7 +42,7 @@ install() {
     [ -n "$(python3 -m pip list --format columns --disable-pip-version-check | grep -w virtualenv)" ] || python3 -m pip install --force-reinstall --user --quiet virtualenv
     [ -d "${virtualenv}" ] || virtualenv --quiet -p python3 ${virtualenv}
     source ${virtualenv}/bin/activate
-    python3 -m pip install --quiet tensorflow
+    python3 -m pip install --quiet --upgrade tensorflow
     python3 -m pip install --quiet ${third_party}/${identifier}
     deactivate
 }

+ 1 - 1
tools/onnx

@@ -55,7 +55,7 @@ install() {
     [ -n "$(python3 -m pip list --format columns --disable-pip-version-check | grep -w virtualenv)" ] || python3 -m pip install --force-reinstall --user --quiet virtualenv
     [ -d "${virtualenv}" ] || virtualenv --quiet -p python3 ${virtualenv}
     source ${virtualenv}/bin/activate
-    python3 -m pip install --quiet protobuf
+    python3 -m pip install --quiet --upgrade protobuf
     export ONNX_ML=1
     export ONNX_NAMESPACE=onnx
     python3 -m pip install --quiet ${third_party}/onnx

+ 4 - 4
tools/pytorch

@@ -45,10 +45,10 @@ install() {
     [ -n "$(python3 -m pip list --format columns --disable-pip-version-check | grep -w virtualenv)" ] || python3 -m pip install --force-reinstall --user --quiet virtualenv
     [ -d "${virtualenv}" ] || python3 -m virtualenv --quiet -p python3 ${virtualenv}
     source ${virtualenv}/bin/activate
-    python3 -m pip install --quiet future protobuf numpy scikit-build scipy
-    python3 -m pip install --quiet ninja pyyaml mkl mkl-include setuptools cmake cffi six typing
-    python3 -m pip install --quiet --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
-    python3 -m pip install --quiet --pre torchvision -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
+    python3 -m pip install --quiet --upgrade future protobuf numpy scikit-build scipy
+    python3 -m pip install --quiet --upgrade ninja pyyaml mkl mkl-include setuptools cmake cffi six typing
+    python3 -m pip install --quiet --pre --upgrade torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
+    python3 -m pip install --quiet --pre --upgrade torchvision -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
     deactivate
 }
 

+ 1 - 1
tools/tf

@@ -51,7 +51,7 @@ install() {
     [ -n "$(python3 -m pip list --format columns --disable-pip-version-check | grep -w virtualenv)" ] || python3 -m pip install --force-reinstall --user --quiet virtualenv
     [ -d "${virtualenv}" ] || virtualenv --quiet -p python3 ${virtualenv}
     source ${virtualenv}/bin/activate
-    python3 -m pip install --quiet protobuf
+    python3 -m pip install --quiet --upgrade protobuf
     deactivate
 }