Improve error handling on missing tritonserver executable

allegroai · May 13, 2021 · 34d4357 · 34d4357
1 parent 435c537
commit 34d4357
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/clearml_serving/triton_helper.py b/clearml_serving/triton_helper.py
@@ -114,7 +114,12 @@ def maintenance_daemon(
             cmd.append('--{}={}'.format(k, v))
 
         print('Starting server: {}'.format(cmd))
-        proc = subprocess.Popen(cmd)
+        try:
+            proc = subprocess.Popen(cmd)
+        except FileNotFoundError:
+            raise ValueError(
+                "Triton Server Engine (tritonserver) could not be found!\n"
+                "Verify you running inside the `nvcr.io/nvidia/tritonserver` docker container")
         base_freq = min(update_frequency_sec, metric_frequency_sec)
         metric_tic = update_tic = time()
         while True: