diff --git a/requirements.txt b/requirements.txt
index a5854fea755d20cc95afd161f587e709a29cfd19..b0a6e7f59896fd0edf08977ee553c803f6c2e960 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -34,6 +34,7 @@ patsy==0.5.1
 Pillow==8.1.0
 pluggy==0.13.1
 protobuf==3.15.0
+psutil==5.8.0
 py==1.10.0
 pydot==1.4.2
 pyparsing==2.4.7
diff --git a/requirements_gpu.txt b/requirements_gpu.txt
index 809eb0b303a745ae9d68dfb5aa059aeebcf24ac6..35fe0d5ee2a03f01737bc185d2a5bbaf26383806 100644
--- a/requirements_gpu.txt
+++ b/requirements_gpu.txt
@@ -34,6 +34,7 @@ patsy==0.5.1
 Pillow==8.1.0
 pluggy==0.13.1
 protobuf==3.15.0
+psutil==5.8.0
 py==1.10.0
 pydot==1.4.2
 pyparsing==2.4.7