Skip to content

Commit e37b967

Browse files
committed
test with tensorflow aarch 64
1 parent e4b0e51 commit e37b967

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

.github/container/pip-finalize.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ else
5454
if [[ "$(uname -m)" = "x86_64" ]]; then
5555
sed -i 's/^tensorflow==\([0-9.*]\+\)$/tensorflow-cpu==\1/' requirements.txt
5656
elif [[ "$(uname -m)" = "arm64" ]] || [[ "$(uname -m)" = "aarch64" ]]; then
57-
sed -i 's/^tensorflow==.*$/tensorflow-metal/' requirements.txt
57+
sed -i 's/^tensorflow==.*$/tensorflow-aarch64/' requirements.txt
5858
else
5959
echo "Skipping TF on $(uname -m)"
6060
fi

0 commit comments

Comments
 (0)