Set torchvision<0.21.0 to match torch/torch_npu version (#479)

Yikun · web-flow · commit 579d858a2075 · 2025-04-08T09:15:42.000+08:00
### What this PR does / why we need it? Set torchvision<0.21.0 to match torch/torch_npu version to resolve `RuntimeError: operator torchvision::nms does not exist`. Closes: #477 ### Does this PR introduce _any_ user-facing change? No ### How was this patch tested? CI passed Signed-off-by: Yikun Jiang <yikunkero@gmail.com>
diff --git a/pyproject.toml b/pyproject.toml
@@ -11,5 +11,6 @@ requires = [
     "setuptools-scm>=8",
     "torch_npu >= 2.5.1rc1",
     "torch >= 2.5.1",
+    "torchvision<0.21.0",
 ]
 build-backend = "setuptools.build_meta"
diff --git a/requirements.txt b/requirements.txt
@@ -9,3 +9,4 @@ setuptools>=64
 setuptools-scm>=8
 torch_npu >= 2.5.1rc1
 torch >= 2.5.1
+torchvision<0.21.0

Original file line number	Diff line number	Diff line change
`@@ -11,5 +11,6 @@ requires = [`
`11`	`11`	`"setuptools-scm>=8",`
`12`	`12`	`"torch_npu >= 2.5.1rc1",`
`13`	`13`	`"torch >= 2.5.1",`
	`14`	`+ "torchvision<0.21.0",`
`14`	`15`	`]`
`15`	`16`	`build-backend = "setuptools.build_meta"`