Update WorkerThread.java

pytorch · Sep 29, 2023 · 722b337 · 722b337
1 parent 4dbd4d0
commit 722b337
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/frontend/server/src/main/java/org/pytorch/serve/wlm/WorkerThread.java b/frontend/server/src/main/java/org/pytorch/serve/wlm/WorkerThread.java
@@ -124,7 +124,7 @@ public String getGpuUsage() {
                         new ProcessBuilder(
                                 "nvidia-smi",
                                 "-i",
-                                String.valueOf(validatedGpuId),
+                                String.valueOf(gpuId),
                                 "--query-gpu=utilization.gpu,utilization.memory,memory.used",
                                 "--format=csv");