Update smoke test

pytorch · Jan 29, 2024 · f9b074d · f9b074d
1 parent f384ffb
commit f9b074d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/packaging/smoke_test.py b/packaging/smoke_test.py
@@ -11,6 +11,8 @@
 for dll in glob.glob(os.path.join(extra_cudnn, "**", "*.dll"), recursive=True):
     shutil.copy(dll, extra_torch)
 
+import torch
+
 print(f"Torch CUDA version: {torch.version.cuda}")
 
 result = subprocess.run(
@@ -23,5 +25,4 @@
 
 
 import tensorrt
-import torch
 import torch_tensorrt