Despite installing the torch vision pytorch library, I am getting an error saying that there is no module named torch vision

Question

The error that I am getting when I use import torchvision is this:

Error Message

"*Traceback (most recent call last):
  File "/Users/gokulsrin/Desktop/torch_basics/data.py", line 4, in <module>
    import torchvision
ModuleNotFoundError: No module named 'torchvision'*"

I don't know what to do. I have tried changing the version of python from the native one to the one downloaded through anaconda. I am using anaconda as a package manager and have installed torch vision through anaconda as well as through pip commands.

Answer 1

From PyTorch installing Docs you should follow these steps:

• In Anaconda use this command:
  conda install pytorch torchvision cpuonly -c pytorch

• In Pip use this command:
  pip3 install torch==1.3.1+cpu torchvision==0.4.2+cpu -f https://download.pytorch.org/whl/torch_stable.html

Note: If you have an enabled CUDA card you can change the cpuonly option to cudatoolkit=10.1 or cudatoolkit=9.2

After successfully installing the package you can import it with the command import torchvision and the output should look like this:

Otherwise, there is something wrong when you are downloading the package from the Internet