diff --git a/test/smoke_test.py b/test/smoke_test.py
index c3a4bdd19d6..65247d22628 100644
--- a/test/smoke_test.py
+++ b/test/smoke_test.py
@@ -1,4 +1,5 @@
-import torch
+"""Run smoke tests"""
+
 import torchvision
-import torchvision.datasets as dset
-import torchvision.transforms
+
+print("torchvision version is ", torchvision.__version__)