Force cgroupfs in TensorDock

dstackai · Oct 30, 2023 · ab7c27a · ab7c27a
1 parent 3bb54a4
commit ab7c27a
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/dstack/_internal/core/backends/tensordock/compute.py b/src/dstack/_internal/core/backends/tensordock/compute.py
@@ -1,3 +1,4 @@
+import json
 from typing import List, Optional
 
 import requests
@@ -73,6 +74,22 @@ def run_job(
                     "runcmd": [
                         ["sh", "-c", " && ".join(commands)],
                     ],
+                    "write_files": [
+                        {
+                            "path": "/etc/docker/daemon.json",
+                            "content": json.dumps(
+                                {
+                                    "runtimes": {
+                                        "nvidia": {
+                                            "path": "nvidia-container-runtime",
+                                            "runtimeArgs": [],
+                                        }
+                                    },
+                                    "exec-opts": ["native.cgroupdriver=cgroupfs"],
+                                }
+                            ),
+                        }
+                    ],
                 },
             )
         except requests.HTTPError: