Fix bug when using int value

mlcommons · Dec 12, 2023 · ef1d2bb · ef1d2bb
1 parent eef8e56
commit ef1d2bb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/runners/mlcube_docker/mlcube_docker/docker_run.py b/runners/mlcube_docker/mlcube_docker/docker_run.py
@@ -276,6 +276,8 @@ def run(self) -> None:
         else:
             cuda_visible_devices = num_gpus
 
+        cuda_visible_devices = str(cuda_visible_devices)
+
         if cuda_visible_devices.isnumeric():
             cuda_visible_devices = str(list(range(cuda_visible_devices)))
             cuda_visible_devices = cuda_visible_devices.replace(" ", "")[1:-1]