Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
vision
Commits
d5929257
Unverified
Commit
d5929257
authored
May 26, 2022
by
Nikita Shulga
Committed by
GitHub
May 26, 2022
Browse files
Add M1 wheels binary builds (#5948)
parent
6aaa2b00
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
48 additions
and
0 deletions
+48
-0
.github/workflows/build-m1-binaries.yml
.github/workflows/build-m1-binaries.yml
+48
-0
No files found.
.github/workflows/build-m1-binaries.yml
0 → 100644
View file @
d5929257
name
:
Build on M1
on
:
pull_request
:
paths
:
-
.github/workflows/build-m1-binaries.yml
push
:
branches
:
-
nightly
workflow_dispatch
:
jobs
:
build_wheels
:
name
:
"
Build
TorchVision
M1
wheels"
runs-on
:
macos-m1
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
:
|
echo $PATH
. ~/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 -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 wheel to GitHub
uses
:
actions/upload-artifact@v3
with
:
name
:
torchvision-py${{ matrix.py_vers }}-macos11-m1
path
:
dist/
-
name
:
Upload wheel to S3
shell
:
arch -arch arm64 bash {0}
env
:
AWS_ACCESS_KEY_ID
:
${{ secrets.AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID }}
AWS_SECRET_ACCESS_KEY
:
${{ secrets.AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY }}
CHANNEL
:
nightly
run
:
|
for pkg in dist/*; do
aws s3 cp "$pkg" "s3://pytorch/whl/${CHANNEL}/cpu/" --acl public-read
done
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment