diff --git a/map2map/test.py b/map2map/test.py
index dc87486..d2b86a1 100644
--- a/map2map/test.py
+++ b/map2map/test.py
@@ -11,6 +11,7 @@ from .utils import import_attr, load_model_state_dict
 
 
 def test(args):
+    print('pytorch {}'.format(torch.__version__))
     pprint(vars(args))
     sys.stdout.flush()
 
diff --git a/map2map/train.py b/map2map/train.py
index 0898563..091364a 100644
--- a/map2map/train.py
+++ b/map2map/train.py
@@ -184,6 +184,7 @@ def gpu_worker(local_rank, node, args):
         logger = SummaryWriter()
 
     if rank == 0:
+        print('pytorch {}'.format(torch.__version__))
         pprint(vars(args))
         sys.stdout.flush()