From 5bf3239f64574558921e2e9d4cf3a411a026f773 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Wed, 26 Feb 2025 14:41:03 +0000 Subject: [PATCH] Test on CUDA 12.8 --- .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 3eebc48fc..65b06e933 100644 --- a/.github/workflows/linux_cuda_wheel.yaml +++ b/.github/workflows/linux_cuda_wheel.yaml @@ -66,7 +66,7 @@ jobs: # For the actual release we should add that label and change this to # include more python versions. python-version: ['3.9'] - cuda-version: ['11.8', '12.6'] + cuda-version: ['11.8', '12.6', '12.8'] # TODO: put back ffmpeg 5 https://github.com/pytorch/torchcodec/issues/325 ffmpeg-version-for-tests: ['4.4.2', '6', '7']