draft: Experimentation in Windows CI

packaging/smoke_test.py

@@ -1,3 +1,36 @@
-import tensorrt
+import glob
+import os
+import shutil
+import site
+import subprocess
+
+python_site_packages = site.getsitepackages()[-1]
+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))}")
+print(f"Current PATH: {os.getenv('PATH')}")
+
+# for dll in glob.glob(os.path.join(extra_cudnn, "**", "*.dll"), recursive=True):
+#     shutil.copy(dll, extra_torch)
+
+# Move Torch cuDNN DLLs into main path
+for dll in glob.glob(os.path.join(extra_torch, "**", "*cudnn*.dll"), recursive=True):
+    shutil.copy(dll, extra_cudnn)
+
 import torch
+
+print(f"Torch CUDA version: {torch.version.cuda}")
+
+result = subprocess.run(
+    ["nvcc", "--version"],
+    capture_output=True,
+    text=True,
+)
+print(result.stdout)
+print(result.stderr)
+
+
+import tensorrt
 import torch_tensorrt