fix: modify docker compose file to use all npus

2024-09-11 16:04:06 +08:00 · 2024-09-11 16:04:06 +08:00 · 0e5781fe6e
parent 283f6f1ab0
commit 0e5781fe6e
1 changed files with 7 additions and 0 deletions
--- a/docker/docker-npu/docker-compose.yml
+++ b/docker/docker-npu/docker-compose.yml
@ -25,6 +25,13 @@ services:
    command: bash
    devices:
      - /dev/davinci0
+      - /dev/davinci1
+      - /dev/davinci2
+      - /dev/davinci3
+      - /dev/davinci4
+      - /dev/davinci5
+      - /dev/davinci6
+      - /dev/davinci7
      - /dev/davinci_manager
      - /dev/devmm_svm
      - /dev/hisi_hdc