Skip to content

Commit 22a8877

Browse files
authored
arm64 build: use same version of numpy as tensorflow-metal (#1328)
1 parent fa8187f commit 22a8877

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

reqs/build.pip

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
numpy<1.20
1+
numpy<1.20; platform_machine != "arm64"
22
protobuf
33
pytest
44
six

scripts/env_create.sh

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -91,6 +91,11 @@ fi
9191
echo "Installing basic build requirements."
9292
if [[ $include_build_deps == 1 ]]; then
9393
python -m pip install -r $COREMLTOOLS_HOME/reqs/build.pip
94+
95+
if [[ $(uname -m) == "arm64" ]]; then
96+
# Install the earliest version of numpy, we can find, that support arm64
97+
conda install -c apple numpy==1.19.5 -y
98+
fi
9499
fi
95100

96101
# Install test requirements (upgrades packages if required)

0 commit comments

Comments
 (0)