added steps to install dependencies for building wheels

pytorch · May 26, 2020 · 0f7b0c2 · 0f7b0c2
1 parent 59943ab
commit 0f7b0c2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/binaries/pip/build_wheels.sh b/binaries/pip/build_wheels.sh
@@ -1,6 +1,7 @@
 #!/usr/bin/env bash
 
 set -eou pipefail
+source ../../scripts/install_from_src_utils
 
 cleanup()
 {
@@ -39,6 +40,10 @@ rm -rf output
 
 mkdir output
 
+install_java_deps
+
+install_torch_deps
+
 create_torchserve_wheel
 
 create_model_archiver_wheel