diff --git a/.github/scripts/install_libs.sh b/.github/scripts/install_libs.sh
index 193aaf46d..b74839150 100644
--- a/.github/scripts/install_libs.sh
+++ b/.github/scripts/install_libs.sh
@@ -28,4 +28,4 @@ elif [ "$CHANNEL" = "test" ]; then
 fi
 
 
-${CONDA_RUN} pip install importlib-metadata click PyYAML
+${CONDA_RUN} pip install importlib-metadata click PyYAML psutil
diff --git a/requirements.txt b/requirements.txt
index 6239d0d90..fc79635f5 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -16,6 +16,7 @@ tqdm
 usort
 parameterized
 PyYAML
+psutil
 
 # for tests
 # https://github.com/pytorch/pytorch/blob/b96b1e8cff029bb0a73283e6e7f6cc240313f1dc/requirements.txt#L3