add sriov in docker stack

2024-07-03 04:30:24 +00:00
parent ef91582d78
commit ec4d6f45c7
1 changed files with 4 additions and 2 deletions
--- a/finetune/docker-compose_stack2.yml
+++ b/finetune/docker-compose_stack2.yml
@@ -48,12 +48,14 @@ services:
            - discrete_resource_spec:
                kind: "NVIDIA-GPU"
                value: 8
+            - discrete_resource_spec:
+                kind: "SRIOV-VF"
+                value: 1
      placement:
        constraints: [node.labels.gpu == true]
    cap_add:
      - IPC_LOCK
-    devices:
-      - "/dev/infiniband:/dev/infiniband:rwm"
+    privileged: true

 networks:
  my-custom-bridge: