Jupyter Notebook 无法识别“导入火炬”

Question

为了在 Ubuntu 上安装PyTorch ，按照官方网站的说明，我做了pip3 install torch torchvision ，我可以使用python3.5命令运行PyTorch 。

但是，当我运行Jupyter Notebook （我只是在终端中运行Jupyter Notebook并使用 Chrome 访问我的笔记本）时，它无法识别包，向我抛出ModuleNotFoundError: No module named 'torch' 。

另一件奇怪的事是PyTorch似乎只安装在Python 3.5而不是Python 3.6上，因为：

➜  ~ python3.5 -c "import torch; print(torch.__version__)"     
0.4.1
➜  ~ python3.6 -c "import torch; print(torch.__version__)"     
Traceback (most recent call last):
  File "<string>", line 1, in <module>
ModuleNotFoundError: No module named 'torch'
➜  ~

因此我猜Jupyter Notebook没有使用Python 3.5 。

这是我在终端中键入python并按TAB时的结果：

➜  ~ python
python             python2.7          python3.5          python3.5m         python3.6          python3.6m         python3-config     python3m-config  
python2            python3            python3.5-config   python3.5m-config  python3.6-config   python3.6m-config  python3m

另外，当我运行which python时，我得到/home/mhy/anaconda3/bin/python因为我最近安装了Anaconda 。 然后我激活了我的PyTorch env

source activate my_env
conda install torch torchvision

但我也无法在Anaconda上运行PyTorch 。

Answer 1

首先，检查您是否在与安装 PyTorch 的环境不同的环境中运行 Jupyter notebook：

(my_env) instance-1:~$ source activate my_env
(my_env) instance-1:~$ python
>>> import  sys
>>> print(sys.executable)
/home/instance-1/anaconda3/envs/my_env/bin/python

然后在my_env环境中运行笔记本： (my_env) instance-1:~$ jupyter notebook并在笔记本中执行print(sys.executable) ，然后如果你得到类似'/home/instance-1/anaconda3/bin/python' ，您已在 Jupyter 默认使用的环境中创建了笔记本。

我一直在努力解决同样的问题，一旦我发现在 Jupyter 笔记本控制台中，您可以选择在不同环境中创建笔记本。 因此，只需转到“新建”部分并选择Python(myenv)之类的内容，即可在所需环境中创建笔记本（本例中为my_env ）。

另一种可能的解决方案可能是将 Jupyter 安装到您的环境中：激活所需的环境并运行conda install -c anaconda jupyter