Skip to content

Commit 478a664

Browse files
committed
Correct bug with cpu count calculation during container exec process
1 parent 93bb63e commit 478a664

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

run_functions.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -312,8 +312,8 @@ def run(self):
312312
if self.use_gpu:
313313
cpu_count = min(multiprocessing.cpu_count(), 5)
314314
else:
315-
cpus = get_text(self.main_gui.ui.SYSTEM__NUM_CPUS__INPUT)
316-
cpus = multiprocessing.cpu_count() if cpus == "All" else int(cpus)
315+
cpu_count = get_text(self.main_gui.ui.SYSTEM__NUM_CPUS__INPUT)
316+
cpu_count = multiprocessing.cpu_count() if cpu_count == "All" else int(cpu_count)
317317
shm_size = f"{128*cpu_count}m"
318318

319319
# dist_backend = "gloo" if self.windows_os else "nccl"

0 commit comments

Comments
 (0)