Skip to content

ModuleNotFoundError: No module named 'cv2' #173

@surfingdoggo

Description

@surfingdoggo

I just installed Anaconda, downloaded the repo, and activated conda, all per docs/the readme. I'm getting this:

(ldm)...$ python scripts/txt2img.py --prompt "a photograph of an astronaut riding a horse" --plms 
Traceback (most recent call last):
  File "/home/c/.code/lab/stable-diffusion/scripts/txt2img.py", line 2, in <module>
    import cv2
ModuleNotFoundError: No module named 'cv2'

Which seems to stem from line 15 in environment.yaml (I guess opencv is cv2):

    - opencv-python==4.1.2.30

When I run conda search opencv-python, I'm getting this:

$ conda search opencv-python
Loading channels: done
No match found for: opencv-python. Search: *opencv-python*

PackagesNotFoundError: The following packages are not available from current channels:

  - opencv-python

Current channels:

  - https://repo.anaconda.com/pkgs/main/linux-64
  - https://repo.anaconda.com/pkgs/main/noarch
  - https://repo.anaconda.com/pkgs/r/linux-64
  - https://repo.anaconda.com/pkgs/r/noarch

To search for alternate channels that may provide the conda package you're
looking for, navigate to

    https://anaconda.org

and use the search bar at the top of the page.

conda list returns a list that seems to have what I need:

opencv-python             4.1.2.30                 pypi_0    pypi
opencv-python-headless    4.6.0.66                 pypi_0    pypi

opencv-python's latest version is 4.6.0.66 as of 8 June 2022.

Solution:
I'm on Kubuntu with Python3.9.
Make sure pip is installed (new install for me)
Make sure distutils are installed and on the latest (apt install python3.9-distutils)
Use pip to install opencv-python (python -m pip install opencv-python)

It seemed to fix my problem. It seems like to use conda to install opencv-python you have to use an unofficial channel which I'm not stoked about, but using pip is another option. Just for detail, here's why I updated distutils.

Now to figure out where torch is at! Happy diffusing.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions