Skip to content

Commit 0c4d714

Browse files
author
Daniel Flores
committed
update cuda wheels to pull from test channel, add back 13.0
1 parent 61dbdf8 commit 0c4d714

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

.github/workflows/docs.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ jobs:
9292
run: ${CONDA_RUN} python -m pip install --upgrade pip
9393
- name: Install PyTorch
9494
run: |
95-
${CONDA_RUN} python -m pip install torch torchvision --index-url https://download.pytorch.org/whl/cu${{ env.cuda_version_without_periods }}
95+
${CONDA_RUN} python -m pip install torch torchvision --index-url https://download.pytorch.org/whl/test/cu${{ env.cuda_version_without_periods }}
9696
${CONDA_RUN} python -c 'import torch; print(f"{torch.__version__}"); print(f"{torch.__file__}"); print(f"{torch.cuda.is_available()=}")'
9797
- name: Install torchcodec from the wheel
9898
run: |

.github/workflows/linux_cuda_wheel.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ jobs:
6666
# the wheel unless the label ciflow/binaries/all is present in the
6767
# PR.
6868
python-version: ['3.9', '3.10', '3.11', '3.12', '3.13']
69-
cuda-version: ['12.6', '12.8', '12.9']
69+
cuda-version: ['12.6', '12.8', '12.9', '13.0']
7070
# TODO: put back ffmpeg 5 https://github.com/pytorch/torchcodec/issues/325
7171
ffmpeg-version-for-tests: ['4.4.2', '5', '6', '7']
7272

@@ -105,7 +105,7 @@ jobs:
105105
run: ${CONDA_RUN} python -m pip install --upgrade pip
106106
- name: Install PyTorch
107107
run: |
108-
${CONDA_RUN} python -m pip install torch torchvision --index-url https://download.pytorch.org/whl/cu${{ env.cuda_version_without_periods }}
108+
${CONDA_RUN} python -m pip install torch torchvision --index-url https://download.pytorch.org/whl/test/cu${{ env.cuda_version_without_periods }}
109109
${CONDA_RUN} python -c 'import torch; print(f"{torch.__version__}"); print(f"{torch.__file__}"); print(f"{torch.cuda.is_available()=}")'
110110
- name: Install torchcodec from the wheel
111111
run: |

0 commit comments

Comments
 (0)