未发现已安装 Jupyter Lab Torch 模块

Question

我安装了 torch 库，当我尝试将其上传到 jupyter lab 时，出现以下错误。有问题的库是 pytorch，在 Python 上它可以正确加载，但是当我在 Jupyter 上尝试时，它告诉我 ModuleNotFoundError: No module named 'torch'

Answer 1

1. 验证 Jupyter 正在使用的 Python 环境

在 Jupyter 笔记本单元中运行此命令：!which python

将输出与安装“torch”的 Python 环境进行比较。

2. 在 Jupyter 环境中安装“torch”

在 Jupyter 笔记本单元中运行此命令：!pip install torch

或者，如果使用“conda”： !conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

3. 确保 Jupyter 内核匹配正确的 Python 环境

如果您希望 Jupyter 使用已安装“torch”的 Python 环境：

在 Python 环境中安装“ipykernel”包：pip install ipykernel

将 Python 环境添加到 Jupyter： python -m ipykernel install --user --name=myenv --display-name "Python (myenv)"

将“myenv”替换为您的 Python 环境的名称。

重新启动 Jupyter Lab 并选择正确的内核：

打开 Jupyter 实验室。 转到“内核”菜单 >“更改内核”> 选择名为“Python (myenv)”的内核。 4.重启Jupyter Lab

确保“torch”安装在正确的环境中或 Jupyter 使用所需的环境后，重新启动 Jupyter Lab 并再次测试。