Skip to content

Commit bd389f4

Browse files
author
SamoraHunter
committed
cpu only torch docs build for git runner
1 parent 0bc585b commit bd389f4

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

.github/workflows/docs.yml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,9 @@ jobs:
2828
NODE_EXTRA_CA_CERTS: /etc/ssl/certs/ca-certificates.crt
2929
SSL_CERT_FILE: /etc/ssl/certs/ca-certificates.crt
3030
REQUESTS_CA_BUNDLE: /etc/ssl/certs/ca-certificates.crt
31+
# Use CPU-only PyTorch to save disk space
32+
PIP_EXTRA_INDEX_URL: https://download.pytorch.org/whl/cpu
33+
TORCH_CPU_ONLY: "true"
3134

3235
steps:
3336
- name: Checkout repository
@@ -64,6 +67,10 @@ jobs:
6467
- name: Install dependencies
6568
run: pip install -e .[docs]
6669

70+
- name: Verify PyTorch installation
71+
run: |
72+
python -c "import torch; print(f'PyTorch version: {torch.__version__}'); print(f'CUDA available: {torch.cuda.is_available()}')"
73+
6774
- name: Build documentation
6875
run: sphinx-build -b html docs/source docs/build
6976

0 commit comments

Comments
 (0)