diff --git a/pkgs/development/python-modules/numba/default.nix b/pkgs/development/python-modules/numba/default.nix
index ecc987156fc1..b0ca879248e8 100644
--- a/pkgs/development/python-modules/numba/default.nix
+++ b/pkgs/development/python-modules/numba/default.nix
@@ -73,8 +73,8 @@ in buildPythonPackage rec {
     cudaPackages.cuda_nvcc
   ];
 
-  buildInputs = with cudaPackages; [
-    cuda_cudart
+  buildInputs = lib.optionals cudaSupport [
+    cudaPackages.cuda_cudart
   ];
 
   propagatedBuildInputs = [