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:
|
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)
|
memory_in_GB = psutil.virtual_memory().total / (1024**3)
|
||||||
|
|
||||||
return min(int(memory_in_GB / thread_memory_in_GB), num_threads)
|
return min(int(memory_in_GB / thread_memory_in_GB), num_threads)
|
||||||
|
Loading…
Reference in New Issue
Block a user