Modulenotfounderror: No module named ‘torch’ (Fix the error)

Modulenotfounderror: No module named ‘torch’ error occurs when PyTorch python module is either not found or its path is not set. Actually, PyTorch is a deep learning library that is compatible with multiple hardware configurations like CPU and CUDA-supported GPUs. The installation process of Pytorch is slightly different for different hardware configurations. Actually, the commands are a bit different. Anyways in this article, we will explore multiple ways to fix this  (No module named ‘torch’) error. So let’s see.

What is Pytorch?

Pytorch is a machine learning framework that is built upon the torch library. Using it you can build applications based on computer vision and natural language processing. The PyTorch framework is open source and built by the Meta company. It is so popular with developers because it supports over 200 different mathematical operations.

Pytorch vs Torch

Many readers get confused about whether pytorch is the same as a torch. The answer is that both are different. Pytorch is a deep learning framework that allows you to build and train deep neural networks. It also provides you a dynamic computational graph for flexibility in the construction and debugging of the model. You can say Pytorch has evolved from the Torch library.

Torch is a scientific computation framework supported by machine learning frameworks. The main coding language for the Torch is Lua and mainly focuses the numerical operations and linear algebra.

Modulenotfounderror: No module named ‘torch’ ( Solution ) –

We can use any Python package manager for the installation of the PyTorch module. Here we will explore pip and conda as package managers or we can use source code to install pytorch package.

Solution 1: Using pip to install pytorch –

Actually, the package manager installs the binary /wheel file at the required path to execute in a single command. Here is the pip command to install PyTorch.

pip install torch

The above command will install the latest version of the PyTorch module. You can specify any other version with the same command.

pip install torch==version

Now if you are using Python 3 then specifically we can use pip3 in the above command.

Other Supplementary packages –

Mostly we do not use any PyTorch alone. Actually, there are a couple of packages we use as a bundle. Here is the command for installing them all together.

pip install torch torchvision torchaudio

Hardware specification with pip –

Now let’s see the hardware configurations with pytorch module. Mostly Deep Learning application works smoothly with  CUDA GPU-supportive library. Here is the command for that.

pip3 install torch==1.10.0+cu113

If you want to install CUDA-supportive supplementary packages. Use the below command.

pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio===0.10.0+cu113 -f

Solution 2: Using conda to install Pytorch –

Conda package manager comes default with Anaconda distribution. Just quite similar to pip we can use the below command.

conda install -c pytorch pytorch

Installing Other Supplementary packages with Conda –

Just like the above package manager pip, we can install additional packages with conda too. Follow the below command for that.

conda install pytorch torchvision torchaudio cpuonly -c pytorch

Hardware specification with conda –

To install the CUDA supportive libraries, Here is the command.

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

Solution 3: Pytorch Installation using source code-

We can download the source code from GitHub and using the setup.py file we can install the same. This is not very straight but one of the ways to solve module not found error.

Download the source code for pytorch.

python setup.py install

Solution 4: Verify the Environment

The other solution for this error is that you should verify whether you are running the code in the torch environment or not. Different environments like conda and virtual have different installation paths. You can check all the installed packages in your system using the below command.

pip

pip list

conda

conda list

Solution 5: Add the Path for Environment variable

There may be a case when you have installed the pytorch module but the environment variable is not set. Thus to resolve the error you will set the path for the Pytorch using the below lines of code.

import sys
sys.path.append('/path/to/pytorch')

Thanks

Data Science Learner Team

Join our list

Subscribe to our mailing list and get interesting stuff and updates to your email inbox.

We respect your privacy and take protecting it seriously

Thank you for signup. A Confirmation Email has been sent to your Email Address.

Something went wrong.