Print all DLLS

pytorch · Jan 29, 2024 · 97eab11 · 97eab11
1 parent f9b074d
commit 97eab11
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/packaging/smoke_test.py b/packaging/smoke_test.py
@@ -8,6 +8,9 @@
 extra_cudnn = os.path.join(python_site_packages, "nvidia", "cudnn")
 extra_torch = os.path.join(python_site_packages, "torch", "lib")
 
+print(f"CUDNN DLLS: {list(glob.glob(os.path.join(extra_cudnn, '**', '*.dll'), recursive=True))}")
+print(f"ALL TORCH DLLS: {list(glob.glob(os.path.join(extra_torch, '**', '*.dll'), recursive=True))}")
+
 for dll in glob.glob(os.path.join(extra_cudnn, "**", "*.dll"), recursive=True):
     shutil.copy(dll, extra_torch)