No module named 'torch' in Jupyter Notebook

Question

I am fairly new to using jupyter notebook, and I've read every forum available for this issue, with no luck.

I am running Windows 11, and I installed Anaconda, then created and activated a virtual environment called pytorchenv. My .yml file includes the following

name: pytorchenv channels:

• defaults
• pytorch dependencies:
• numpy=1.16.2
• pandas=0.24.2
• matplotlib=3.0.3
• pillow=5.4.1
• pip=19.0
• plotly=3.7.0
• scikit-learn=0.20.3
• seaborn=0.9.0
• python=3.7.3
• jupyter=1.0.0
• pytorch=1.1.0
• torchvision=0.2.2

If I list all of the included packages in the command prompt, using

conda list -n pytorch

it shows that pytorch is installed as...

Name                    Version                   Build  Channel
pytorch                   1.1.0           py3.7_cuda100_cudnn7_1    pytorch

What's more, if I enable the virtual environment in the Command Prompt, It seems that I am able to import pytorch successfully

C:\\Users\\Xxxx\>conda activate pytorchenv

(pytorchenv) C:\\Users\\Xxxx\>python
Python 3.7.3 (default, Apr 24 2019, 15:29:51) \[MSC v.1915 64 bit (AMD64)\] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>>import torch
>>>

However, when I try to import pytorch into jupyter notebook, I get the message

import torch
ModuleNotFoundError                       
Traceback (most recent call last)
Input In \[1\], in \<cell line: 1\>()
1 import torch
ModuleNotFoundError: No module named 'torch'

I have been able to import pandas and numpy in jupyter notebook without issue, but I have not figured out a way to import pytorch successfully.

In case this is helpful, jupyter notebook shows that it is running on

Python 3 (ipykernel)

and running the following in jupyter notebook

import sys
print(sys.executable)

results in

C:\\Users\\Nathaniel\\anaconda3\\python.exe

I've tried checking if there were multiple versions of jupyter notebook on my computer (there aren't).

I also checked that pytorch is installed in the same place as my other anaconda packages (it is).

What am I doing wrong?

Answer 1

The notebook is running with your base environment, as indicated by sys.executable.

Two methods:

• Start Jupyter from your environment
• Register your environment to be usable by all jupyter notebooks
  For that you can find the necessary steps here

Basic step is this

source activate pytorchenv   # or activate the environment via conda
python -m ipykernel install --user --name pytorchenv --display-name "Python (pytorchenv)"

Afterwards be sure to restart Juypter, then you should be able to select your pytorchenv kernel.