Skip to content

Commit a00c1b5

Browse files
authored
Update linux_cuda_wheel.yaml
1 parent 911afa9 commit a00c1b5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.github/workflows/linux_cuda_wheel.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ jobs:
7474
ffmpeg-version-for-tests: ['4.4.2', '6', '7', '8.0']
7575

7676
container:
77-
image: "pytorch/manylinux2_28-builder:cuda${{ matrix.cuda-version }}"
77+
image: "pytorch/manylinux2_28-builder:cuda${{ matrix.cuda-version }}-33c74927083180f050d839e75420d4554d986dee"
7878
options: "--gpus all -e NVIDIA_DRIVER_CAPABILITIES=video,compute,utility"
7979
needs: build
8080
steps:
@@ -108,7 +108,7 @@ jobs:
108108
run: ${CONDA_RUN} python -m pip install --upgrade pip
109109
- name: Install PyTorch
110110
run: |
111-
${CONDA_RUN} python -m pip install --pre torch==2.10.0.dev20251106 torchvision --index-url https://download.pytorch.org/whl/nightly/cu${{ env.cuda_version_without_periods }}
111+
${CONDA_RUN} python -m pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu${{ env.cuda_version_without_periods }}
112112
${CONDA_RUN} python -c 'import torch; print(f"{torch.__version__}"); print(f"{torch.__file__}"); print(f"{torch.cuda.is_available()=}")'
113113
- name: Install torchcodec from the wheel
114114
run: |

0 commit comments

Comments
 (0)