calls setup_kubernetes before _setup to set cudaDeviceCount

Duke-GCB · May 2, 2024 · f225c44 · f225c44
1 parent b943289
commit f225c44
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/calrissian/job.py b/calrissian/job.py
@@ -709,9 +709,11 @@ def get_pod_name(pod):
             self.make_tmpdir()
         self.populate_env_vars(runtimeContext)
 
-        self._setup(runtimeContext)
         # specific setup for Kubernetes
         self.setup_kubernetes(runtimeContext)
+
+        self._setup(runtimeContext)
+
         pod = self.create_kubernetes_runtime(runtimeContext) # analogous to create_runtime()
         self.execute_kubernetes_pod(pod) # analogous to _execute()
         completion_result = self.wait_for_kubernetes_pod()