diff --git a/pkgs/development/python-modules/paddlex/default.nix b/pkgs/development/python-modules/paddlex/default.nix
index 68f196c3dcf1..a60c3e1d8534 100644
--- a/pkgs/development/python-modules/paddlex/default.nix
+++ b/pkgs/development/python-modules/paddlex/default.nix
@@ -17,6 +17,7 @@
   ruamel-yaml,
   typing-extensions,
   ujson,
+  distutils,
 }:
 
 let
@@ -34,6 +35,10 @@ let
 
     build-system = [ setuptools ];
 
+    dependencies = [ distutils ];
+
+    pythonImportsCheck = [ "GPUtil" ];
+
     meta = {
       homepage = "https://github.com/anderskm/gputil";
       license = lib.licenses.mit;