【发布时间】:2019-02-11 18:41:58
【问题描述】:
我在名为 pytorch1 的 Conda 环境中按照 PyTorch 1.0 的安装说明进行操作:
(pytorch1) me@comp:~$ conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
号称安装成功,日志显示pytorch-1.0.1和python-3.7.2,但是我打开ipython时却遇到了以下情况:
Python 3.6.6 |Anaconda, Inc.| (default, Jun 28 2018, 17:14:51)
Type 'copyright', 'credits' or 'license' for more information
IPython 6.5.0 -- An enhanced Interactive Python. Type '?' for help.
In [1]: import torch
In [2]: torch.__version__
Out[2]: '0.4.1.post2'
这当然是 PyTorch 的错误版本。
我可以打开 python3.7 并安装正确的 PyTorch:
(pytorch1) me@comp:~$ python3.7
Python 3.7.2 (default, Dec 29 2018, 06:19:36)
[GCC 7.3.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.__version__
'1.0.1.post2'
我想在我当前的 Jupyter 笔记本设置中使用 PyTorch。有没有办法让它工作,这样当我在pytorch1 环境中打开 iPython 时,我可以import torch 使版本 1 可用?
更新:我更新了 Anaconda,现在 Python 版本正确但 iPython 找不到 PyTorch
【问题讨论】:
-
conda install ipython在您使用的 conda 环境中应该解决这个问题吗? -
@UmangGupta 成功了——我真的很感激。如果您将其放在答案中,我会将其标记为“最佳答案”