Docs:Supplement NPU environment installation document

2026-03-21 16:18:13 +00:00 · 2025-12-30 15:57:13 +08:00
parent 8f1d10fb43
commit 9cc1697d4d
3 changed files with 30 additions and 8 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -34,5 +34,17 @@ classifiers = [

 [tool.setuptools.packages.find]

+[project.optional-dependencies]
+npu_aarch64 = [
+    "torch==2.7.1",
+    "torch-npu==2.7.1",
+    "torchvision==0.22.1"
+]
+npu = [
+    "torch==2.7.1+cpu",
+    "torch-npu==2.7.1",
+    "torchvision==0.22.1+cpu"
+]
+
 [tool.setuptools]
 include-package-data = true