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并运行它。如果一切设置正确,这应该不会报错。
声明
本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。