setup.sh 875 B

123456789101112131415161718192021222324252627282930313233
  1. #!/bin/bash
  2. set -ex -o pipefail
  3. echo ""
  4. echo "DIR: $(pwd)"
  5. TEST_APP_PATH=$(dirname $(realpath $0))
  6. cd ${TEST_APP_PATH}
  7. PYTORCH_IOS_NIGHTLY_NAME=libtorch_ios_nightly_build.zip
  8. VISION_IOS_NIGHTLY_NAME=libtorchvision_ops_ios_nightly_build.zip
  9. echo "Downloading torch libs and vision libs..."
  10. wget https://ossci-ios-build.s3.amazonaws.com/${PYTORCH_IOS_NIGHTLY_NAME}
  11. wget https://ossci-ios-build.s3.amazonaws.com/${VISION_IOS_NIGHTLY_NAME}
  12. mkdir -p ./library/torch
  13. mkdir -p ./library/vision
  14. echo "Unziping torch libs and vision libs..."
  15. unzip -d ./library/torch ./${PYTORCH_IOS_NIGHTLY_NAME}
  16. unzip -d ./library/vision ./${VISION_IOS_NIGHTLY_NAME}
  17. cp ./library/vision/install/lib/*.a ./library/torch/install/lib
  18. cp -r ./library/torch/install .
  19. rm -rf ./library
  20. rm -rf ./*.zip
  21. echo "Generating the vision model..."
  22. python ./make_assets.py
  23. echo "Finished project setups."