diff --git a/.github/workflows/test-integration.yml b/.github/workflows/test-integration.yml
index 44a2abbf..e2afa334 100644
--- a/.github/workflows/test-integration.yml
+++ b/.github/workflows/test-integration.yml
@@ -34,7 +34,7 @@ jobs:
         run: uv python install 3.13
 
       - name: Install dependencies
-        run: uv sync --group dev
+        run: uv sync --group dev --extra torch-cu128
 
       - name: Run integration tests
         env: