diff --git a/test/smoke_test.py b/test/smoke_test.py
index e2a3b5068ab..f66ff8df54a 100644
--- a/test/smoke_test.py
+++ b/test/smoke_test.py
@@ -1,6 +1,10 @@
 """Run smoke tests"""
 
 import os
+os.environ["CUDA_LOG_FILE"] = "stdout"
+os.environ["TORCH_SHOW_CPP_STACKTRACES"] = "1"
+import subprocess
+print(subprocess.check_output(["nvidia-smi"], text=True))
 import sys
 import sysconfig
 from pathlib import Path