Skip to content

Commit 468fa82

Browse files
authored
Use torch 2.9 release packages (#15168)
### Summary Use release 2.9 packages for pytorch and domain libraries in preparation for the final 1.0 release. I'm following the pattern from #13248.
1 parent 2d68b88 commit 468fa82

File tree

5 files changed

+7
-28
lines changed

5 files changed

+7
-28
lines changed

.ci/docker/common/install_pytorch.sh

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,12 +12,11 @@ source "$(dirname "${BASH_SOURCE[0]}")/utils.sh"
1212

1313
install_domains() {
1414
echo "Install torchvision and torchaudio"
15-
pip_install --no-use-pep517 --user "git+https://github.com/pytorch/audio.git@${TORCHAUDIO_VERSION}"
16-
pip_install --no-use-pep517 --user "git+https://github.com/pytorch/vision.git@${TORCHVISION_VERSION}"
15+
pip install torchvision==0.24.0 torchaudio==2.9.0 --index-url https://download.pytorch.org/whl/cpu
1716
}
1817

1918
install_pytorch_and_domains() {
20-
pip_install torch==2.9.0 torchvision torchaudio torchao==0.14.0 --index-url https://download.pytorch.org/whl/test/cpu
19+
pip_install torch==2.9.0 torchvision==0.24.0 torchaudio==2.9.0 torchao==0.14.0 --index-url https://download.pytorch.org/whl/cpu
2120
}
2221

2322
install_pytorch_and_domains

.ci/scripts/setup-linux.sh

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -17,9 +17,9 @@ echo "Build tool: $BUILD_TOOL, Mode: $BUILD_MODE"
1717
# have already been installed, so we use PyTorch build from source here instead
1818
# of nightly. This allows CI to test against latest commits from PyTorch
1919
if [[ "${EDITABLE:-false}" == "true" ]]; then
20-
install_executorch --use-pt-pinned-commit --editable
20+
install_executorch --editable
2121
else
22-
install_executorch --use-pt-pinned-commit
22+
install_executorch
2323
fi
2424
build_executorch_runner "${BUILD_TOOL}" "${BUILD_MODE}"
2525

@@ -28,7 +28,3 @@ build_executorch_runner "${BUILD_TOOL}" "${BUILD_MODE}"
2828
if [ "$(uname -m)" == "aarch64" ]; then
2929
conda install -y -c conda-forge libstdcxx-ng
3030
fi
31-
32-
if [[ "${GITHUB_BASE_REF:-}" == *main* || "${GITHUB_BASE_REF:-}" == *gh* ]]; then
33-
do_not_use_nightly_on_ci
34-
fi

install_requirements.py

Lines changed: 0 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -147,25 +147,6 @@ def install_requirements(use_pytorch_nightly):
147147

148148

149149
def install_optional_example_requirements(use_pytorch_nightly):
150-
print("Installing torch domain libraries")
151-
DOMAIN_LIBRARIES = [
152-
("torchvision==0.24.0" if use_pytorch_nightly else "torchvision"),
153-
"torchaudio==2.9.0" if use_pytorch_nightly else "torchaudio",
154-
]
155-
# Then install domain libraries
156-
subprocess.run(
157-
[
158-
sys.executable,
159-
"-m",
160-
"pip",
161-
"install",
162-
*DOMAIN_LIBRARIES,
163-
"--extra-index-url",
164-
TORCH_URL,
165-
],
166-
check=True,
167-
)
168-
169150
print("Installing packages in requirements-examples.txt")
170151
subprocess.run(
171152
[

pyproject.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,7 @@ dependencies=[
7979
"hydra-core>=1.3.0",
8080
"omegaconf>=2.3.0",
8181
"torchao==0.14.0",
82+
"torch>=2.9.0,<2.10.0",
8283
]
8384

8485
[project.urls]

requirements-examples.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,3 +5,5 @@ timm == 1.0.7
55
torchsr == 1.0.4
66
torchtune >= 0.6.1
77
transformers == 4.53.1
8+
torchaudio == 2.9.0
9+
torchvision == 0.24.0

0 commit comments

Comments
 (0)