test cudnn overwrite 2

pytorch · Jan 26, 2024 · e0366e5 · e0366e5
1 parent 41ae59b
commit e0366e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packaging/smoke_test.py b/packaging/smoke_test.py
@@ -8,7 +8,7 @@
 extra_torch = os.path.join(python_site_packages, "torch", "lib")
 
 for dll in glob.glob(os.path.join(extra_cudnn, "**", "*.dll"), recursive=True):
-    shutil.copy(extra_cudnn, extra_torch)
+    shutil.copy(dll, extra_torch)
 
 
 # import tensorrt