没有名为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安装火炬torchvision但是收到错误。

Answer 1

这对我有用：

1. 创建一个conda虚拟环境：

    conda create -n env_pytorch python=3.6 

2. 活跃于此环境创建以上：

    source activate env_pytorch 

3. 使用pip或pip3安装PyTorch：

    pip install torchvision --user