Use site.getsitepackages()[-1] for both Windows and Linux.

microsoft · Dec 11, 2024 · 02c73c6 · 02c73c6
1 parent d2cbf27
commit 02c73c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxruntime/__init__.py b/onnxruntime/__init__.py
@@ -92,7 +92,7 @@
     import site
 
     # Get the site-packages path where nvidia packages are installed
-    site_packages_path = site.getsitepackages()[0]
+    site_packages_path = site.getsitepackages()[-1]
     nvidia_path = os.path.join(site_packages_path, "nvidia")
     # Traverse the directory and subdirectories
     if platform.system() == "Windows":  #