Hey Mr. Bing,
If you use env:// as the init_method, it would look for MASTER_ADDR and MASTER_PORT env vars. You can use sth like init_method='tcp://ip:port' or directly set the env var. E.g.:
    os.environ['MASTER_ADDR'] = 'localhost'
    os.environ['MASTER_PORT'] = '29500'