Skip to content

Commit f7e1e74

Browse files
HolyWugs-olive
authored andcommitted
cuDNN is not required by TRT 10
1 parent 3ffcdb0 commit f7e1e74

File tree

1 file changed

+1
-23
lines changed

1 file changed

+1
-23
lines changed

packaging/smoke_test_windows.py

Lines changed: 1 addition & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -1,25 +1,6 @@
1-
import glob
2-
import os
3-
import shutil
4-
import site
51
import subprocess
62

7-
python_site_packages = site.getsitepackages()[-1]
8-
extra_cudnn = os.path.join(python_site_packages, "nvidia", "cudnn")
9-
extra_torch = os.path.join(python_site_packages, "torch", "lib")
10-
11-
print(
12-
f"CUDNN DLLS: {list(glob.glob(os.path.join(extra_cudnn, '**', '*.dll'), recursive=True))}"
13-
)
14-
print(
15-
f"ALL TORCH DLLS: {list(glob.glob(os.path.join(extra_torch, '**', '*.dll'), recursive=True))}"
16-
)
17-
print(f"Current PATH: {os.getenv('PATH')}")
18-
19-
# Move TRT cuDNN DLLs into torch path
20-
for dll in glob.glob(os.path.join(extra_cudnn, "**", "*.dll"), recursive=True):
21-
shutil.copy(dll, extra_torch)
22-
3+
import tensorrt # noqa: F401
234
import torch
245

256
print(f"Torch CUDA version: {torch.version.cuda}")
@@ -31,6 +12,3 @@
3112
)
3213
print(result.stdout)
3314
print(result.stderr)
34-
35-
36-
import tensorrt # noqa: F401

0 commit comments

Comments
 (0)