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

Question

当我发出Anaconda提示符conda search 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

用通过Conda安装PyTorch写的，但没有用。顺便说一句，安装Anaconda后是否需要重新启动机器？另外，让我知道如何在木星笔记本中使用PyTorch。

Answer 1

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

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

如果上面在木星笔记本中抛出了同样的问题，如果您已经启用了GPU，请尝试重新启动木星笔记本服务器，因为有时它需要重新启动，用户报告。

当我试图从木星笔记本中导入这个包时，我收到了以下错误消息: ModuleNotFoundError:没有名为“火炬”的模块。然后，我尝试从Anaconda导航器为我的环境安装木星笔记本应用程序(Torch)。重新启动我的木星笔记本，运行进口火炬，这一次它起作用了

否则，您只需要CPU版本的PyTorch。

conda install pytorch torchvision cpuonly -c pytorch