Нет модуля с именем torch.distributed

Question

ImportError: Ни один модуль с именем torch.distributed

File "train.py", line 4, in <module>
    import torch.distributed as dist
ImportError: No module named torch.distributed

Я установил CUDA И cuDNN, затем создал env и установил pip3 install torchvision, но получил ошибку.

Answer 1

Это работает для меня:

1. Создание виртуальной среды conda:

   conda create -n env_pytorch python=3.6
   

2. Активируйте эту среду, создайте выше:

   source activate env_pytorch
   

3. Установить PyTorch с pip или pip3:

   pip install torchvision --user