Skip to content

Commit 805ea2d

Browse files
authored
Update torch versions for 0.3 release (#3986)
1 parent d27c3a4 commit 805ea2d

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

install_requirements.sh

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -62,12 +62,12 @@ done
6262
NIGHTLY_VERSION=dev20240507
6363

6464
# The pip repository that hosts nightly torch packages.
65-
TORCH_NIGHTLY_URL="https://download.pytorch.org/whl/nightly/cpu"
65+
TORCH_URL="https://download.pytorch.org/whl/test/cpu"
6666

6767
# pip packages needed by exir.
6868
EXIR_REQUIREMENTS=(
69-
torch=="2.4.0.${NIGHTLY_VERSION}"
70-
torchvision=="0.19.0.${NIGHTLY_VERSION}" # For testing.
69+
torch=="2.4.0"
70+
torchvision=="0.19.0" # For testing.
7171
)
7272

7373
# pip packages needed for development.
@@ -85,7 +85,7 @@ DEVEL_REQUIREMENTS=(
8585
# TODO(dbort): Make each example publish its own requirements.txt
8686
EXAMPLES_REQUIREMENTS=(
8787
timm==0.6.13
88-
torchaudio=="2.2.0.${NIGHTLY_VERSION}"
88+
torchaudio=="2.4.0"
8989
torchsr==1.0.4
9090
transformers==4.38.2
9191
)
@@ -100,7 +100,7 @@ REQUIREMENTS_TO_INSTALL=(
100100

101101
# Install the requirements. `--extra-index-url` tells pip to look for package
102102
# versions on the provided URL if they aren't available on the default URL.
103-
$PIP_EXECUTABLE install --extra-index-url "${TORCH_NIGHTLY_URL}" \
103+
$PIP_EXECUTABLE install --extra-index-url "${TORCH_URL}" \
104104
"${REQUIREMENTS_TO_INSTALL[@]}"
105105

106106
#

0 commit comments

Comments
 (0)