fix torchvision::nms not exist (#5671)

This commit is contained in:
Yineng Zhang
2025-04-23 02:17:21 -07:00
committed by GitHub
parent 8777a1d24b
commit 127ff8982e

View File

@@ -88,7 +88,7 @@ jobs:
- name: Install
run: |
bash scripts/ci_install_dependency.sh
pip3 install torch==2.6.0 && pip3 install pytest
pip3 install torch==2.6.0 torchvision && pip3 install pytest
pip3 uninstall sgl-kernel -y || true
pip3 install sgl-kernel/dist/*whl --force-reinstall --no-deps
pip3 list | grep sgl-kernel