File tree Expand file tree Collapse file tree 1 file changed +48
-0
lines changed Expand file tree Collapse file tree 1 file changed +48
-0
lines changed Original file line number Diff line number Diff line change 1+ name : Build on M1
2+ on :
3+ pull_request :
4+ paths :
5+ - .github/workflows/build-m1-binaries.yml
6+ push :
7+ branches :
8+ - nightly
9+ workflow_dispatch :
10+ jobs :
11+ build_wheels :
12+ name : " Build TorchVision M1 wheels"
13+ runs-on : macos-m1
14+ strategy :
15+ matrix :
16+ py_vers : [ "3.8", "3.9", "3.10" ]
17+ steps :
18+ - name : Checkout repository
19+ uses : actions/checkout@v2
20+ - name : Build TorchVision M1 wheel
21+ shell : arch -arch arm64 bash {0}
22+ env :
23+ ENV_NAME : conda-env-${{ github.run_id }}
24+ PY_VERS : ${{ matrix.py_vers }}
25+ run : |
26+ echo $PATH
27+ . ~/miniconda3/etc/profile.d/conda.sh
28+ set -ex
29+ conda create -yp ${ENV_NAME} python=${PY_VERS} numpy libpng openjpeg wheel pkg-config
30+ conda run -p ${ENV_NAME} python3 -mpip install torch --extra-index-url=https://download.pytorch.org/whl/nightly torch
31+ conda run -p ${ENV_NAME} python3 -mpip install delocate
32+ conda run -p ${ENV_NAME} python3 setup.py bdist_wheel
33+ conda env remove -p ${ENV_NAME}
34+ - name : Upload wheel to GitHub
35+ uses : actions/upload-artifact@v3
36+ with :
37+ name : torchvision-py${{ matrix.py_vers }}-macos11-m1
38+ path : dist/
39+ - name : Upload wheel to S3
40+ shell : arch -arch arm64 bash {0}
41+ env :
42+ AWS_ACCESS_KEY_ID : ${{ secrets.AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID }}
43+ AWS_SECRET_ACCESS_KEY : ${{ secrets.AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY }}
44+ CHANNEL : nightly
45+ run : |
46+ for pkg in dist/*; do
47+ aws s3 cp "$pkg" "s3://pytorch/whl/${CHANNEL}/cpu/" --acl public-read
48+ done
You can’t perform that action at this time.
0 commit comments