Skip to content

Commit 5e61df8

Browse files
add link for stable release torch package until pypi download is available
1 parent d4daa6a commit 5e61df8

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

packaging/pre_build_script_arm64.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,8 @@ pip install numpy==2.2.3
5858

5959
if [ "$CHANNEL" = "release" ]; then
6060
echo "Installing latest stable version of PyTorch."
61-
pip3 install torch
61+
# TODO: update when arm64 torch available on pypi
62+
pip3 install --pre torch --index-url https://download.pytorch.org/whl/torch/
6263
elif [ "$CHANNEL" = "test" ]; then
6364
echo "Installing PyTorch version $PYTORCH_VERSION."
6465
pip3 install torch=="$PYTORCH_VERSION"

0 commit comments

Comments
 (0)