Invoke wheel cleaner after the install step #13
@ -39,7 +39,7 @@ def disk_free_in_gb(builder: worker.utils.Builder) -> float:
|
||||
|
||||
|
||||
def get_thread_count(thread_memory_in_GB: float) -> int:
|
||||
num_threads = psutil.cpu_count()
|
||||
num_threads = psutil.cpu_count() or 1
|
||||
memory_in_GB = psutil.virtual_memory().total / (1024**3)
|
||||
|
||||
return min(int(memory_in_GB / thread_memory_in_GB), num_threads)
|
||||
|
Loading…
Reference in New Issue
Block a user