Can not import torch when use python -m torch.distributed.launch

1

I have already installed pytorch, version=1.0, when distributed training, it seems that node can not import torch? What should I do?

python -m torch.distributed.launch --nproc_per_node=2 main_fp16_optimizer.py -a resnet50 --fp16 --b 224 --workers 4

Traceback (most recent call last):
  File "main_fp16_optimizer.py", line 6, in <module>
    import torch
ImportError: No module named torch
Traceback (most recent call last):
  File "main_fp16_optimizer.py", line 6, in <module>
    import torch
ImportError: No module named torch
2

are you sure that pytorch is installed?
can you try import torch in python terminal?

3

PyTorch not installed properly!

4

It seems I have not imported python path into environment path.