os.environ['RANK'] keyError: 'RANK'
I run
python main.py --mode train --data data1/ImageNetOrigin --config config/mbv2/retrain.yaml
--flops 43 --chcfg ./results/DMCPMobileNetV2_43_MMDDHH/model_sample/expected_ch
Traceback (most recent call last):
File "main.py", line 75, in
Turn off distributed training
I have a machine with 8 GPUs and 24 CPUs. Why doesn't this key exists in os.environ?
$ python
Python 3.8.5 (default, Sep 4 2020, 07:30:14)
[GCC 7.3.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import os
>>> os.environ["RANK"]
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/anaconda/envs/azureml_py38/lib/python3.8/os.py", line 675, in __getitem__
raise KeyError(key) from None
KeyError: 'RANK'
I specifically want to do distributed training.