logo标签列表

无法在Jupyter笔记本中导入torch

pythonjupyter-notebookpytorch
10

系统：macOS 10.13.6 Python 版本：3.7 Anaconda3

在 jupyter notebook 中使用 import torch 出现问题。

ModuleNotFoundError: No module named 'torch'

以下是我安装 PyTorch 的步骤：

conda install pytorch torchvision -c pytorch

我已经检查过，PyTorch已经安装在我的anaconda环境中:

图片描述输入在此

当我在终端中键入python3并且import torch是可以运行的。但是在jupyter notebook中无法工作。

我已经尝试过：

conda update conda
conda install mkl=2018

但仍然出现相同的错误。

1
在安装PyTorch之前，您应该先安装Jupyter Notebook。您的安装顺序是什么？ - one
你在Jupyter Notebook中使用不同的内核吗（Kernel -> 更改内核）？ - yascho
@Alpha 在安装PyTorch之前，我先安装了Jupyter Notebook。 - Shin Yu Wu
@yascho 我有Python2和Python3。但是当我新建一个Jupyter笔记本时，我只能选择Python3，因为打开Python2时会显示内核错误。 - Shin Yu Wu
8个回答
9

在您激活的conda环境中，除了安装pytorch，还需要安装jupyter。以下是安装步骤:

1. 创建conda环境

例如: 使用Python 3.7创建名为pytorch_p37的环境:

user@pc:~$ conda create -n pytorch_p37 python=3.7

2. 激活它

user@pc:~$ conda activate pytorch_p37

或者对于较旧的 conda 版本，可以使用以下命令：
user@pc:~$ source activate pytorch_p37

现在你应该在命令提示符前看到 (pytorch_p37)。
(pytorch_p37) user@pc:~$

3. 前往PyTorch网站，通过conda选择适当的安装命令。在你的shell中运行它，例如：

(pytorch_p37) user@pc:~$ conda install pytorch torchvision -c pytorch

4. 将jupyter安装到你激活的环境中

(pytorch_p37) user@pc:~$ conda install jupyter

5. 验证安装

(pytorch_p37) user@pc:~$ conda list

# packages in environment at /home/user/anaconda3/envs/pytorch_p37:
#
# Name
...
jupyter                   1.0.0  
jupyter_client            5.3.1  
jupyter_console           6.0.0 
jupyter_core              4.5.0
...
python                    3.7.4
pytorch                   1.2.0
...

6. 运行 jupyter


(pytorch_p37) user@pc:~$ jupyter notebook
谢谢你的回复！我按照你的步骤操作，但仍然出现相同的错误 :( - Shin Yu Wu
嗯... conda install mkl=2018 怎么样？我没有添加这行代码，你可以试试吗？ - trsvchn
还是出现了同样的错误。在pytorch_p37环境中，我输入：python --version，结果显示Python 3.7.4而不是Python 3.6.3 |Anaconda custom (64-bit)| (default, Oct 13 2017, 12:02:49)<-- 我从另一个网站复制过来的。这有关系吗？ - Shin Yu Wu
Python看起来没问题。conda update conda可能是一个conda的问题，如果你有时间和耐心 :) 你可以尝试conda update conda，然后重复所有步骤（也包括mkl）。 - trsvchn
2

我曾经遇到过同样的问题，在虚拟环境中成功安装了torch，但jupyter lab无法导入。但是当我使用conda list查看已安装的包列表时，发现PyTorch已经安装，但是jupyterlab没有安装。安装了jupyterlab后，问题得到解决！

你是对的！谢谢。通过确保我的环境处于活动状态，然后安装torch，然后Jupyiter笔记本修复了“找不到torch模块”的问题。 - Niyongabo Eric
0

我刚刚通过在Anaconda Powershell提示符上运行conda install pytorch torchvision -c pytorch解决了这个问题。

实际结果

(base) PS C:\Users\TayyabChaudhary> conda install pytorch torchvision -c pytorch
Collecting package metadata (current_repodata.json): done
Solving environment: done

## Package Plan ##

  environment location: C:\Users\TayyabChaudhary\anaconda3

  added / updated specs:
    - pytorch
    - torchvision


The following packages will be downloaded:

    package                    |            build
    ---------------------------|-----------------
    ca-certificates-2021.9.30  |       haa95532_1         116 KB
    certifi-2021.10.8          |   py38haa95532_0         152 KB
    conda-4.10.3               |   py38haa95532_0         2.9 MB
    cudatoolkit-11.3.1         |       h59b6b97_2       545.3 MB
    libuv-1.40.0               |       he774522_0         255 KB
    openssl-1.1.1l             |       h2bbff1b_0         4.8 MB
    pytorch-1.10.0             |py3.8_cuda11.3_cudnn8_0        1.44 GB  pytorch
    pytorch-mutex-1.0          |             cuda           3 KB  pytorch
    torchvision-0.11.1         |       py38_cu113         8.8 MB  pytorch
    ------------------------------------------------------------
                                           Total:        1.99 GB

The following NEW packages will be INSTALLED:

  cudatoolkit        pkgs/main/win-64::cudatoolkit-11.3.1-h59b6b97_2
  libuv              pkgs/main/win-64::libuv-1.40.0-he774522_0
  pytorch            pytorch/win-64::pytorch-1.10.0-py3.8_cuda11.3_cudnn8_0
  pytorch-mutex      pytorch/noarch::pytorch-mutex-1.0-cuda
  torchvision        pytorch/win-64::torchvision-0.11.1-py38_cu113

The following packages will be UPDATED:

  ca-certificates    anaconda::ca-certificates-2020.10.14-0 --> pkgs/main::ca-certificates-2021.9.30-haa95532_1
  certifi                anaconda::certifi-2020.6.20-py38_0 --> pkgs/main::certifi-2021.10.8-py38haa95532_0
  conda                               4.10.1-py38haa95532_1 --> 4.10.3-py38haa95532_0
  openssl                                 1.1.1k-h2bbff1b_0 --> 1.1.1l-h2bbff1b_0


Proceed ([y]/n)? y


Downloading and Extracting Packages
libuv-1.40.0         | 255 KB    | ############################################################################ | 100%
torchvision-0.11.1   | 8.8 MB    | ############################################################################ | 100%
certifi-2021.10.8    | 152 KB    | ############################################################################ | 100%
cudatoolkit-11.3.1   | 545.3 MB  | ############################################################################ | 100%
openssl-1.1.1l       | 4.8 MB    | ############################################################################ | 100%
pytorch-1.10.0       | 1.44 GB   | ############################################################################ | 100%
conda-4.10.3         | 2.9 MB    | ############################################################################ | 100%
ca-certificates-2021 | 116 KB    | ############################################################################ | 100%
pytorch-mutex-1.0    | 3 KB      | ############################################################################ | 100%
Preparing transaction: done
Verifying transaction: done
Executing transaction: done
0

问题已经解决。我的环境是Windows10，conda 4.8.3和Python 3.8.3。

以下是我所做的：

  1. 进入Anaconda提示符shell并激活一个Anaconda环境。我的环境是PyTorch。

activate pytorch

  1. 在Anaconda提示符shell中为Jupyter笔记本创建一个内核，以将Anaconda环境链接到Jupyter笔记本的内核。

python -m ipykernel install --user --name pytorch --display-name "pytorch"

第一个“pytorch”是Anaconda环境的名称，第二个是Jupyter笔记本的内核名称。确保已安装ipykernel。

3.运行Jupyter笔记本，在步骤2中刚创建的内核中选择它，然后导入torch模块以查看结果。

输入图像描述

0

输入图像描述个人而言，我尝试了不同的解决方案并解决了我的问题。 首先，进入Anaconda提示符并使用命令conda install nb_conda。 其次，进入pytorch的环境并使用conda install ipykernel。 之后，在jupyter笔记本中，我们可以找到更多的语言来使用。选择语言Python [conda env:conda-pytorch]，然后我们就可以成功地使用pytorch运行代码。 希望这能帮到你！

0

我也遇到了同样的错误。我在Anaconda命令提示符上尝试了以下命令行。

(pytorch)C:\Users\user> conda install jupyter

然后，我使用了命令C:\Users\user> jupyter notebook启动了Jupyter笔记本，import torch关键字开始工作了。

0

我知道创建虚拟环境是一种理想的方法，但通过在VSCode中安装jupyter笔记本扩展程序帮助我解决了这个问题。我不需要创建虚拟环境（这取决于用户是否需要创建）

无关信息。只要Jupyter使用相同的Python，无论是默认Python还是安装了Torch的环境，它都可以工作。 - eval
0

对我来说，通过重新创建虚拟环境，在虚拟环境中使用conda安装jupyterlab，并在jupyter lab会话中运行pip install而不是conda，这种方法可以解决问题。

conda deactivate [venv_name]
python -m ipykernel install --user --name=[venv_name]
conda remove -n [venv_name] --all
conda create -n [venv_name] python=3.9
conda activate [venv_name]
conda install jupyterlab
jupyter lab
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

（根据CUDA版本等，检查PyTorch的特定命令。） https://pytorch.org/get-started/locally/

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文，
原文链接