changed to use free memory instead of total memory

saching13 · saching13 · commit 3e39c73d5190 · 2021-03-18T15:12:39.000-07:00
diff --git a/depthai-core b/depthai-core
@@ -1 +1 @@
-Subproject commit 41cbbec4efda39eab17ecccb38f6c6a43bb139d6
+Subproject commit 0a42076af386e5933eaee0d01fec2c635191418f
diff --git a/setup.py b/setup.py
@@ -117,14 +117,14 @@ def build_extension(self, ext):
         build_args += ['--config', cfg]
 
         # Memcheck (guard if it fails)
-        totalMemory = 4000
+        freeMemory = 4000
         if platform.system() == "Linux":
             try:
-                totalMemory = int(os.popen("free -m").readlines()[1].split()[1])
+                freeMemory = int(os.popen("free -m").readlines()[1].split()[2])
             except (KeyboardInterrupt, SystemExit):
                 raise
             except:
-                totalMemory = 4000
+                freeMemory = 4000
         # Memcheck (guard if it fails)
 
 
@@ -151,10 +151,10 @@ def build_extension(self, ext):
                 os.environ['_PYTHON_HOST_PLATFORM'] = re.sub(r'macosx-[0-9]+\.[0-9]+-(.+)', r'macosx-10.9-\1', util.get_platform())
 
             # Specify how many threads to use when building, depending on available memory
-            if totalMemory < 1000:
+            if freeMemory < 1000:
                 build_args += ['--', '-j1']
                 cmake_args += ['-DHUNTER_JOBS_NUMBER=1']
-            elif totalMemory < 2000:
+            elif freeMemory < 2500:
                 build_args += ['--', '-j2']
                 cmake_args += ['-DHUNTER_JOBS_NUMBER=2']
             else:
