Skip to content

Commit de1e83a

Browse files
committed
Fix Windows site-packages search for CTK 13
1 parent f6ebdbc commit de1e83a

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

‎cuda_pathfinder/cuda/pathfinder/_dynamic_libs/find_nvidia_dynamic_lib.py‎

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,10 @@ def _find_dll_under_dir(dirpath: str, file_wild: str) -> Optional[str]:
5959
def _find_dll_using_nvidia_bin_dirs(
6060
libname: str, lib_searched_for: str, error_messages: list[str], attachments: list[str]
6161
) -> Optional[str]:
62-
nvidia_sub_dirs_list: list[tuple[str, ...]] = [("nvidia", "*", "bin")]
62+
nvidia_sub_dirs_list: list[tuple[str, ...]] = [
63+
("nvidia", "*", "bin"), # CTK 12
64+
("nvidia", "*", "bin", "*"), # CTK 13, e.g. site-packages\nvidia\cu13\bin\x86_64\
65+
]
6366
if libname == "nvvm":
6467
nvidia_sub_dirs_list.append(("nvidia", "*", "nvvm", "bin")) # Only for CTK 12
6568
for nvidia_sub_dirs in nvidia_sub_dirs_list:

0 commit comments

Comments
 (0)