在名为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
答案 0 :(得分:0)
似乎您正在使用的ipython
是使用python3.6
并在conda
环境之外安装的。您可以使用which ipython
来查看它的位置,并猜测它是否与系统python或conda一起安装。
如果使用系统python安装,则只需在conda环境中使用ipython
将conda
与conda install ipython
一起安装即可。那应该解决这个问题。