[caffe2][tpx] log to stderr (pytorch#42162)

Summary: Pull Request resolved: pytorch#42162 Test Plan: CI Reviewed By: mrshenli Differential Revision: D22791440 fbshipit-source-id: 14f16cd7a94a57161c5724177b518527f486232d
zmmwl · Jul 28, 2020 · e179966 · e179966
1 parent 0571cfd
commit e179966
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/distributed/test_nccl.py b/test/distributed/test_nccl.py
@@ -18,7 +18,8 @@
 
 nGPUs = torch.cuda.device_count()
 if not TEST_CUDA:
-    print('CUDA not available, skipping tests')
+    import sys
+    print('CUDA not available, skipping tests', file=sys.stderr)
     TestCase = object  # noqa: F811