Import torch 报错,称找不到模块的解决办法

cyynlaaa 2024-08-22 10:05:10 阅读 63

问题:pytorch已经安装好了。在anaconda的命令窗口提示符下,输入Python,再运行import torch ,可以正确导入模块。但是,在运行jupyter notebook后,import torch 提示找不到模块。为什么?

当你遇到在Anaconda环境中通过命令行可以成功导入PyTorch模块,但在Jupyter Notebook中却提示找不到模块的情况时,这通常是因为Jupyter Notebook没有使用正确的Python环境或Kernel。这里有几个步骤可以帮助你解决这个问题:

确认Jupyter Notebook的Kernel:

打开Jupyter Notebook,点击右上角的“Kernel”菜单,选择“Change kernel”。

检查是否有与你安装PyTorch相同的Anaconda环境对应的Kernel。如果没有,你需要创建一个新的Kernel。

创建新的Kernel:

首先,确保你的PyTorch环境是活跃的。在Anaconda Prompt中,使用conda activate your_env_name(将your_env_name替换为你的环境名)来激活环境。

安装IPython内核到你的环境中,以便Jupyter Notebook可以使用它。在激活的环境中运行以下命令:

conda install ipykernel

python -m ipykernel install --user --name your_env_name --display-name "Python (your_env_name)"

这将在你的Jupyter Notebook中创建一个新的Kernel,名为“Python (your_env_name)”。

重启Jupyter Notebook:

关闭当前的Jupyter Notebook(如果它正在运行)。

重新打开Anaconda Prompt,激活你的环境,然后再次启动Jupyter Notebook:

conda activate your_env_name

jupyter notebook

在Jupyter Notebook中选择新的Kernel:

打开你的Notebook,点击右上角的“Kernel”菜单,选择“Change kernel”,然后选择你刚创建的“Python (your_env_name)”Kernel。

实际情况是:右上角没有这个菜单。可以新建notebook,在箭头处做选择。

尝试导入PyTorch:

在Notebook的一个新单元格中,输入import torch并运行它。如果一切设置正确,这应该不会报错。



声明

本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。