python - 尽管安装了torch vision pytorch 库，但我收到一条错误消息，指出没有名为torch vision 的模块

Question

我使用 import torchvision 时遇到的错误这是:

错误信息

"*Traceback (most recent call last):
  File "/Users/gokulsrin/Desktop/torch_basics/data.py", line 4, in <module>
    import torchvision
ModuleNotFoundError: No module named 'torchvision'*"

我不知道该怎么办。我尝试将 python 的版本从 native 版本更改为通过 anaconda 下载的版本。我使用 anaconda 作为包管理器，并通过 anaconda 以及 pip 命令安装了 torch vision。

Answer 1

来自 PyTorch安装 Docs您应该按照以下步骤操作:

在 python 使用这个命令:
conda install pytorch torchvision cpuonly -c pytorch

在 点使用这个命令:
pip3 install torch==1.3.1+cpu torchvision==0.4.2+cpu -f https://download.pytorch.org/whl/torch_stable.html

注意:如果您启用了 CUDA 卡，您可以更改 cpuonly cudatoolkit=10.1 的选项或 cudatoolkit=9.2成功安装软件包后，您可以使用命令 import torchvision 将其导入。输出应如下所示:

不然你从网上下载包的时候会出错