Skip to content

Commit 33d7edf

Browse files
authored
Install CPU-only PyTorch for GitHub Actions (#2810)
Co-authored-by: Masato Shinokawa <[email protected]>
1 parent a0d0335 commit 33d7edf

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

.github/workflows/docs-build.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ jobs:
2626
env:
2727
ARTIFACT_DAYS: 0
2828
PYPI_LOCAL_DIR: "pypi_pkgs/"
29+
TORCH_URL: "https://download.pytorch.org/whl/cpu"
2930
steps:
3031
- uses: actions/checkout@v5
3132
- uses: actions/setup-python@v6
@@ -39,7 +40,7 @@ jobs:
3940
run: sudo apt-get install -y pandoc
4041
- name: Install package & dependencies
4142
timeout-minutes: 20
42-
run: pip install . -U -r requirements/docs.txt
43+
run: pip install . -U -r requirements/docs.txt --extra-index-url=${TORCH_URL}
4344

4445
- name: Make ${{ matrix.target }}
4546
working-directory: docs/

0 commit comments

Comments
 (0)