diff --git a/imperative/python/megengine/data/dataloader.py b/imperative/python/megengine/data/dataloader.py
index 3d671c1e9c9299070450bbb7bb6a2e748e152fe4..1c09a255d704b68aef68ebce674e53bcaac422ff 100644
--- a/imperative/python/megengine/data/dataloader.py
+++ b/imperative/python/megengine/data/dataloader.py
@@ -184,9 +184,11 @@ class DataLoader:
             logger.warning(
                 "Each worker need to read the shared meta-data, which will be increasing the reference count."
                 "Copy-On-Write propety will lead to 'memory leak', the memory usage will end up being "
-                + total_memory
-                + " GB"
-                "However the current requested memory is " + current_memory + " GB"
+                + str(total_memory)
+                + " GB, "
+                "However the current requested memory is "
+                + str(current_memory)
+                + " GB, "
                 "Maybe you can request more memory or uesd np-array to save meta-data rather than List or Tuple"
             )
 
diff --git a/imperative/python/requires.txt b/imperative/python/requires.txt
index 894b332a02b9c87ac9c57085db6712120fc18f6b..d08566d3f7ef84629ef3cb5e80a5a75ab4e2ba17 100644
--- a/imperative/python/requires.txt
+++ b/imperative/python/requires.txt
@@ -9,3 +9,4 @@ mprop
 wheel
 megfile>=0.0.10
 filelock
+psutil
\ No newline at end of file