fix torchvision::nms not exist (#5671)
This commit is contained in:
2
.github/workflows/pr-test-sgl-kernel.yml
vendored
2
.github/workflows/pr-test-sgl-kernel.yml
vendored
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user