1 files changed, 10 insertions, 0 deletions
diff --git a/examples/graph_vgg16.cpp b/examples/graph_vgg16.cpp
index 4b5f33a7e8..482aab1683 100644
--- a/examples/graph_vgg16.cpp
+++ b/examples/graph_vgg16.cpp
@@ -41,6 +41,16 @@ public:
     }
     bool do_setup(int argc, char **argv) override
     {
+        // Check if the system has enough RAM to run the example, systems with less than 2GB have
+        // to hint the API to minimize memory consumption otherwise it'll run out of memory and
+        // fail throwing the bad_alloc exception
+        arm_compute::MEMInfo meminfo;
+        const size_t         mem_total = meminfo.get_total_in_kb();
+        if(mem_total <= arm_compute::MEMInfo::TWO_GB_IN_KB)
+        {
+            arm_compute::MEMInfo::set_policy(arm_compute::MemoryPolicy::MINIMIZE);
+        }
+
         // Parse arguments
         cmd_parser.parse(argc, argv);