命令行 python 和 jupyter notebook 使用两个不同版本的 torch

Question

在我的 conda 环境中，从命令行 Python 和 jupyter 笔记本导入 torch 会产生两个不同的结果。

命令行 Python：

$ source activate GNN
(GNN) $ python
>>> import torch
>>> print(torch.__file__)
/home/riccardo/.local/lib/python3.7/site-packages/torch/__init__.py
>>> print(torch.__version__)
0.4.1

木星：

(GNN) $ jupyter notebook --no-browser --port=8890

import torch

print(torch.__file__)
/home/riccardo/.local/lib/python3.6/site-packages/torch/__init__.py

print(torch.__version__)
1.2.0+cu92

我尝试了在 Jupyter Notebook 中未显示的 Conda 环境中建议的步骤

$ conda install ipykernel
$ source activate GNN
(GNN) $ python -m ipykernel install --user --name GNN --display-name "Python (GNN)"
Installed kernelspec GNN in /home/riccardo/.local/share/jupyter/kernels/gnn

但这并没有解决问题。

Answer 1

您需要使用 Jupyter 识别 Anaconda 环境

conda activate myenv
conda install -n myenv ipykernel
python -m ipykernel install --user --name myenv --display-name "Python (myenv)"

将myenv替换为您的环境名称。 稍后，在您的 Jupyter Notebook 中，在 Select Kernel 选项中，您将看到此Python (myenv)选项。