[Feat] upgrade pytorch2.6 (#5417)

2025-04-21 07:06:34 +08:00
parent 475e2e378a
commit 417b44eba8
7 changed files with 8 additions and 8 deletions
--- a/.github/workflows/pr-test-sgl-kernel.yml
+++ b/.github/workflows/pr-test-sgl-kernel.yml
@@ -88,7 +88,7 @@ jobs:
      - name: Install
        run: |
          bash scripts/ci_install_dependency.sh
-          pip3 install torch==2.5.1 && pip3 install pytest
+          pip3 install torch==2.6.0 && pip3 install pytest
          pip3 uninstall sgl-kernel -y || true
          pip3 install sgl-kernel/dist/*whl --force-reinstall --no-deps
          pip3 list | grep sgl-kernel