安装PyTorch后jupyter notebook中仍出现“No module named torch“

2022-05-09 09:16:02 浏览数 (1)

安装好 PyTorch1.5.0 之后,在 Anaconda Prompt 中激活新创建的环境变量之后,import torch 并 打印 torch 的版本没有问题,说明 PyTorch 已经安装成功。但是打开 Jupyter Notebook 后 import torch 却提示 “No module named torch”

报错如下:

我们还需要在这个环境安装 jupyter notebook 的一系列依赖库

代码语言:javascript复制
conda install jupyter

接着在这个环境变量中安装nb_conda(关键

代码语言:javascript复制
conda install nb_conda

重启 jupyter notebook 后再进入,选择对应的 Kernel 新建文件即可:

import torch 可以正常使用了

0 人点赞