我遵循了PyTorch 1.0的安装说明,在一个名为pytorch1的Conda环境中
(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'我想使用PyTorch与我目前木星笔记本设置。是否有一种方法可以使其工作,以便当我在pytorch1环境中打开pytorch1时,我可以通过import torch使版本1可用?
更新:我更新了Anaconda,现在iPython版本是正确的,但是iPython找不到PyTorch
发布于 2019-02-12 21:24:56
您使用的ipython似乎是使用python3.6和conda环境外部安装的。您可以使用which ipython查看它的位置,并猜测它是与system或conda一起安装的。
如果它是使用system安装的,您可以在conda环境中使用ipython使用conda install ipython安装conda。这应该能解决这个问题。
https://stackoverflow.com/questions/54637077
复制相似问题