From 84fc517c75ff68dc73f87e5a4a8a9378443e30f5 Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Mon, 10 Nov 2025 16:09:07 -0500 Subject: [PATCH 1/8] Try pinning pytest --- .github/workflows/linux_cuda_wheel.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/linux_cuda_wheel.yaml b/.github/workflows/linux_cuda_wheel.yaml index 17f18fe8b..2ac5c091b 100644 --- a/.github/workflows/linux_cuda_wheel.yaml +++ b/.github/workflows/linux_cuda_wheel.yaml @@ -122,7 +122,7 @@ jobs: - name: Install test dependencies run: | # Ideally we would find a way to get those dependencies from pyproject.toml - ${CONDA_RUN} python -m pip install numpy pytest pillow + ${CONDA_RUN} python -m pip install numpy pytest==8.4.2 pillow - name: Delete the src/ folder just for fun run: | From 911afa98d2ad0543e9200e2adf8491a7ded1ad48 Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Tue, 11 Nov 2025 10:29:01 -0500 Subject: [PATCH 2/8] Update linux_cuda_wheel.yaml --- .github/workflows/linux_cuda_wheel.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/linux_cuda_wheel.yaml b/.github/workflows/linux_cuda_wheel.yaml index 2ac5c091b..6aa84dc77 100644 --- a/.github/workflows/linux_cuda_wheel.yaml +++ b/.github/workflows/linux_cuda_wheel.yaml @@ -108,7 +108,7 @@ jobs: run: ${CONDA_RUN} python -m pip install --upgrade pip - name: Install PyTorch run: | - ${CONDA_RUN} python -m pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu${{ env.cuda_version_without_periods }} + ${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 }} ${CONDA_RUN} python -c 'import torch; print(f"{torch.__version__}"); print(f"{torch.__file__}"); print(f"{torch.cuda.is_available()=}")' - name: Install torchcodec from the wheel run: | From a00c1b5234878a3289c0bf3f90d30bf0b4756498 Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Tue, 11 Nov 2025 11:24:39 -0500 Subject: [PATCH 3/8] Update linux_cuda_wheel.yaml --- .github/workflows/linux_cuda_wheel.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/linux_cuda_wheel.yaml b/.github/workflows/linux_cuda_wheel.yaml index 6aa84dc77..257182dde 100644 --- a/.github/workflows/linux_cuda_wheel.yaml +++ b/.github/workflows/linux_cuda_wheel.yaml @@ -74,7 +74,7 @@ jobs: ffmpeg-version-for-tests: ['4.4.2', '6', '7', '8.0'] container: - image: "pytorch/manylinux2_28-builder:cuda${{ matrix.cuda-version }}" + image: "pytorch/manylinux2_28-builder:cuda${{ matrix.cuda-version }}-33c74927083180f050d839e75420d4554d986dee" options: "--gpus all -e NVIDIA_DRIVER_CAPABILITIES=video,compute,utility" needs: build steps: @@ -108,7 +108,7 @@ jobs: run: ${CONDA_RUN} python -m pip install --upgrade pip - name: Install PyTorch run: | - ${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 }} + ${CONDA_RUN} python -m pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu${{ env.cuda_version_without_periods }} ${CONDA_RUN} python -c 'import torch; print(f"{torch.__version__}"); print(f"{torch.__file__}"); print(f"{torch.cuda.is_available()=}")' - name: Install torchcodec from the wheel run: | From 26dda4827e90744b778cec725f3ee9e17cd67a2c Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Thu, 13 Nov 2025 11:43:46 -0500 Subject: [PATCH 4/8] Update linux_cuda_wheel.yaml --- .github/workflows/linux_cuda_wheel.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/linux_cuda_wheel.yaml b/.github/workflows/linux_cuda_wheel.yaml index 257182dde..d5a278746 100644 --- a/.github/workflows/linux_cuda_wheel.yaml +++ b/.github/workflows/linux_cuda_wheel.yaml @@ -108,7 +108,7 @@ jobs: run: ${CONDA_RUN} python -m pip install --upgrade pip - name: Install PyTorch run: | - ${CONDA_RUN} python -m pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu${{ env.cuda_version_without_periods }} + ${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 }} ${CONDA_RUN} python -c 'import torch; print(f"{torch.__version__}"); print(f"{torch.__file__}"); print(f"{torch.cuda.is_available()=}")' - name: Install torchcodec from the wheel run: | From d9ff8bf09c8465a335cc2e6378f2d32c8ffe1aa2 Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Thu, 13 Nov 2025 13:48:22 -0500 Subject: [PATCH 5/8] Update linux_cuda_wheel.yaml --- .github/workflows/linux_cuda_wheel.yaml | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/.github/workflows/linux_cuda_wheel.yaml b/.github/workflows/linux_cuda_wheel.yaml index d5a278746..991a12ee2 100644 --- a/.github/workflows/linux_cuda_wheel.yaml +++ b/.github/workflows/linux_cuda_wheel.yaml @@ -74,7 +74,7 @@ jobs: ffmpeg-version-for-tests: ['4.4.2', '6', '7', '8.0'] container: - image: "pytorch/manylinux2_28-builder:cuda${{ matrix.cuda-version }}-33c74927083180f050d839e75420d4554d986dee" + image: "pytorch/manylinux2_28-builder:cuda${{ matrix.cuda-version }}" options: "--gpus all -e NVIDIA_DRIVER_CAPABILITIES=video,compute,utility" needs: build steps: @@ -108,7 +108,7 @@ jobs: run: ${CONDA_RUN} python -m pip install --upgrade pip - name: Install PyTorch run: | - ${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 }} + ${CONDA_RUN} python -m pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu${{ env.cuda_version_without_periods }} ${CONDA_RUN} python -c 'import torch; print(f"{torch.__version__}"); print(f"{torch.__file__}"); print(f"{torch.cuda.is_available()=}")' - name: Install torchcodec from the wheel run: | @@ -140,6 +140,7 @@ jobs: ls - name: Run Python tests run: | + export LD_LIBRARY_PATH=$CONDA_PREFIX/lib:/usr/local/cuda/lib64/:${LD_LIBRARY_PATH} ${CONDA_RUN} FAIL_WITHOUT_CUDA=1 pytest --override-ini="addopts=-v" test --tb=short - name: Run Python benchmark run: | From f2c0de06a02da057fbb280cd5f5b47944b397e23 Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Thu, 13 Nov 2025 17:26:15 -0500 Subject: [PATCH 6/8] Update linux_cuda_wheel.yaml --- .github/workflows/linux_cuda_wheel.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/linux_cuda_wheel.yaml b/.github/workflows/linux_cuda_wheel.yaml index 991a12ee2..1e841d615 100644 --- a/.github/workflows/linux_cuda_wheel.yaml +++ b/.github/workflows/linux_cuda_wheel.yaml @@ -84,6 +84,7 @@ jobs: echo cuda_version_without_periods=${cuda_version_without_periods} >> $GITHUB_ENV python_version_without_periods=$(echo "${{ matrix.python-version }}" | sed 's/\.//g') echo python_version_without_periods=${python_version_without_periods} >> $GITHUB_ENV + echo LD_LIBRARY_PATH=$CONDA_PREFIX/lib:/usr/local/cuda/lib64/:${LD_LIBRARY_PATH} >> $GITHUB_ENV - uses: actions/download-artifact@v4 with: name: meta-pytorch_torchcodec__${{ matrix.python-version }}_cu${{ env.cuda_version_without_periods }}_x86_64 @@ -140,7 +141,6 @@ jobs: ls - name: Run Python tests run: | - export LD_LIBRARY_PATH=$CONDA_PREFIX/lib:/usr/local/cuda/lib64/:${LD_LIBRARY_PATH} ${CONDA_RUN} FAIL_WITHOUT_CUDA=1 pytest --override-ini="addopts=-v" test --tb=short - name: Run Python benchmark run: | From b6ab7f99d87598fd24366b337da42002c4f75351 Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Fri, 14 Nov 2025 09:34:14 -0500 Subject: [PATCH 7/8] Update linux_cuda_wheel.yaml --- .github/workflows/linux_cuda_wheel.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/linux_cuda_wheel.yaml b/.github/workflows/linux_cuda_wheel.yaml index 1e841d615..fb8d296a5 100644 --- a/.github/workflows/linux_cuda_wheel.yaml +++ b/.github/workflows/linux_cuda_wheel.yaml @@ -84,7 +84,6 @@ jobs: echo cuda_version_without_periods=${cuda_version_without_periods} >> $GITHUB_ENV python_version_without_periods=$(echo "${{ matrix.python-version }}" | sed 's/\.//g') echo python_version_without_periods=${python_version_without_periods} >> $GITHUB_ENV - echo LD_LIBRARY_PATH=$CONDA_PREFIX/lib:/usr/local/cuda/lib64/:${LD_LIBRARY_PATH} >> $GITHUB_ENV - uses: actions/download-artifact@v4 with: name: meta-pytorch_torchcodec__${{ matrix.python-version }}_cu${{ env.cuda_version_without_periods }}_x86_64 @@ -102,6 +101,7 @@ jobs: ${CONDA_RUN} conda info ${CONDA_RUN} nvidia-smi ${CONDA_RUN} conda list + echo LD_LIBRARY_PATH=$CONDA_PREFIX/lib:/usr/local/cuda/lib64/:${LD_LIBRARY_PATH} >> $GITHUB_ENV - name: Assert ffmpeg exists run: | ${CONDA_RUN} ffmpeg -buildconf @@ -123,7 +123,7 @@ jobs: - name: Install test dependencies run: | # Ideally we would find a way to get those dependencies from pyproject.toml - ${CONDA_RUN} python -m pip install numpy pytest==8.4.2 pillow + ${CONDA_RUN} python -m pip install numpy pytest pillow - name: Delete the src/ folder just for fun run: | From 4794ca60add04a0389af492544125628b5834a5f Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Fri, 14 Nov 2025 09:35:06 -0500 Subject: [PATCH 8/8] Update linux_cuda_wheel.yaml --- .github/workflows/linux_cuda_wheel.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/linux_cuda_wheel.yaml b/.github/workflows/linux_cuda_wheel.yaml index fb8d296a5..4e711b8c3 100644 --- a/.github/workflows/linux_cuda_wheel.yaml +++ b/.github/workflows/linux_cuda_wheel.yaml @@ -95,7 +95,7 @@ jobs: # We install conda packages at the start because otherwise conda may have conflicts with dependencies. # Note: xorg-libxau was addded to fix a problem with ffmpeg 4. We should consider removing it. default-packages: "nvidia/label/cuda-${{ matrix.cuda-version }}.0::libnpp nvidia::cuda-nvrtc=${{ matrix.cuda-version }} nvidia::cuda-toolkit=${{ matrix.cuda-version }} nvidia::cuda-cudart=${{ matrix.cuda-version }} nvidia::cuda-driver-dev=${{ matrix.cuda-version }} conda-forge::ffmpeg=${{ matrix.ffmpeg-version-for-tests }} conda-forge::xorg-libxau" - - name: Check env + - name: Check env, set LD_LIBRARY_PATH run: | ${CONDA_RUN} env ${CONDA_RUN} conda info