Added possible out of shared memory error message (#25730)

pbelevich · facebook-github-bot · commit 5f9cbfa1d635 · 2019-09-14T05:27:48.000-07:00
Summary: Fixes #5040 Pull Request resolved: #25730 Differential Revision: D17226214 Pulled By: pbelevich fbshipit-source-id: 92278272aab74e6690f14fc9597acfd1a98854b7
diff --git a/torch/csrc/DataLoader.cpp b/torch/csrc/DataLoader.cpp
@@ -131,6 +131,10 @@ static PyObject *THPModule_errorIfAnyWorkerFails(PyObject *module) {
         std::ostringstream oss;
         oss << "DataLoader worker (pid " << worker_pid << ") is killed "
             << "by signal: " << strsignal(infop.si_status) << ". ";
+        if (infop.si_status == SIGBUS) {
+            oss << "It is possible that dataloader's workers are out of shared memory. "
+                << "Please try to raise your shared memory limit.";
+        }
         // This is necessary. Otherwise, the runtime error will kill the other
         // workers, and trigger this again.
         pid_set->clear();
