已通过conda安装pytorch，但无法在Windows 10中导入

Question

当我发出Python提示时conda search pytorch 然后我安装了pytorch，即使发出conda list命令也会给出：

pytorch                   1.5.1           py3.7_cuda102_cudnn7_0    pytorch

但是当我在命令提示符下启动python，然后发出import pytorch时，我得到ModuleNotFoundError: No module named 'pytorch'

甚至我在发布后也尝试过这样做

conda create -n pytorch_env -c pytorch pytorch torchvision
conda activate pytorch_env
conda install -c pytorch pytorch torchvision

如Installing PyTorch via Conda中所述，但没有任何用处。顺便问一下，安装Anaconda后是否需要重新启动机器？另外，让我知道如何在Jupyter笔记本中使用Pytork

Answer 1

使用import torch而不是pytorch验证安装。下面的示例代码，source

from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)

如果上述问题在Jupyter笔记本中出现，并且您已经启用了GPU，请尝试重新启动Jupyter笔记本服务器，因为有时需要重新启动，user reported

when I tried to import this package from Jupyter notebook, I got following error message: ModuleNotFoundError: No module named 'torch'. Then, I tried installing Jupyter notebook application from Anaconda navigator for my environment(torch). Restarted my Jupyter notebook and ran import torch and this time it worked

否则，您需要仅CPU版本的PyTorch

conda install pytorch torchvision cpuonly -c pytorch