找不到 Pytorch 模块

Pytorch module not found

我正在使用 Python 3.6,并尝试使用 Pytorch。我已经使用 pip3 卸载了它,然后重新安装了它

Hugos-MacBook-Pro-2:project hugokitano$ pip3 install torch 
Requirement already satisfied: torch in /usr/local/lib/python3.7/site-packages (1.3.1) 
Requirement already satisfied: numpy in /Users/hugokitano/Library/Python/3.7/lib/python/site-packages (from torch) (1.17.2)

但是,当我尝试导入 torch 时,找不到该模块。我也尝试通过 conda 安装,运行

conda install pytorch torchvision -c pytorch

安装成功,安装到

environment location: /Users/hugokitano/anaconda

但是,"import torch"还是不行。有什么想法吗?谢谢!

尝试更新conda并安装它。它奇怪地解决了我的问题。

conda update conda

明智的做法是使用环境而不是在您的基础上安装软件包。尝试执行以下操作:

conda create -n deep7 -c pytorch python=3.7 pytorch torchvision
conda activate deep7
python -c "import torch"

我们创建了一个名为 deep7 的环境,我们使用 pytorch 通道在 Python 3.7 中安装 pytorch。之后我们激活环境并测试导入是否有效。如果是这样,您将看不到任何错误。

要使用 pytorch,您必须激活您的环境: conda activate deep7 并停用 conda deactivate。您可以使用 conda install -n deep7 <package name>

添加库

编码愉快

顺便说一句:如果你想要 Python 3.6,做同样的事情,将上面的所有 7 更改为 6 :)