diff --git a/.github/workflows/build-on-m1.yml b/.github/workflows/build-on-m1.yml new file mode 100644 index 00000000000..c97119735cc --- /dev/null +++ b/.github/workflows/build-on-m1.yml @@ -0,0 +1,33 @@ +name: Build on M1 +on: + pull_request +jobs: + build_vision: + runs-on: macos-m1-11 + strategy: + matrix: + py_vers: [ "3.8", "3.9", "3.10" ] + steps: + - name: Checkout repository + uses: actions/checkout@v2 + - name: Build TorchVision M1 wheel + shell: arch -arch arm64 bash {0} + env: + ENV_NAME: conda-env-${{ github.run_id }} + PY_VERS: ${{ matrix.py_vers }} + run: | + source ~/miniconda3/etc/profile.d/conda.sh + set -ex + conda create -yp ${ENV_NAME} python=${PY_VERS} numpy libpng openjpeg wheel pkg-config + conda run -p ${ENV_NAME} python3 -c "import os;print(os.__file__)" + conda run -p ${ENV_NAME} python3 -mpip install torch --extra-index-url=https://download.pytorch.org/whl/nightly torch + conda run -p ${ENV_NAME} python3 -mpip install delocate + conda run -p ${ENV_NAME} python3 setup.py bdist_wheel + conda env remove -p ${ENV_NAME} + - name: Upload TorchVision wheel + uses: seemethere/upload-artifact-s3@v4 + with: + name: torchvision-py${{ matrix.py_vers }}-macos11-m1 + path: dist/ + s3-bucket: gha-artifacts + region: us-east-1