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
|
- name: Install
|
||||||
run: |
|
run: |
|
||||||
bash scripts/ci_install_dependency.sh
|
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 uninstall sgl-kernel -y || true
|
||||||
pip3 install sgl-kernel/dist/*whl --force-reinstall --no-deps
|
pip3 install sgl-kernel/dist/*whl --force-reinstall --no-deps
|
||||||
pip3 list | grep sgl-kernel
|
pip3 list | grep sgl-kernel
|
||||||
|
|||||||
Reference in New Issue
Block a user