How to print out logs with LOG(INFO) in Python in PyTorch1.10

TORCH_DISTRIBUTED_DEBUG=INFO should work.