add linux specific logic

NVIDIA · Nov 26, 2024 · 8f3c5f7 · 8f3c5f7
1 parent 094e841
commit 8f3c5f7
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/numba_cuda/numba/cuda/cuda_paths.py b/numba_cuda/numba/cuda/cuda_paths.py
@@ -314,10 +314,11 @@ def get_conda_include_dir():
 
 def get_system_include_dir():
     """Return the system CUDA include directory, if it exists"""
-    system_cuda_include = '/usr/local/cuda/include'
-    if os.path.exists(system_cuda_include):
-        return system_cuda_include
-    return None
+    if sys.platform.startswith('linux'):
+        system_cuda_include = '/usr/local/cuda/include'
+        if os.path.exists(system_cuda_include):
+            return system_cuda_include
+    return
 
 
 def _get_include_dir():