ModuleNotFoundError When Importing open_clip

user_17092942 0
Viewed 527

error importing open_clip, When importing open_clip I always get this error:

Traceback (most recent call last):
  File "c:\Users\noahs\Desktop\New folder\test.py", line 1, in <module>
    import open_clip
  File "C:\Users\noahs\Desktop\New folder\folder\lib\site-packages\open_clip\__init__.py", line 2, in <module>
    from .loss import ClipLoss
  File "C:\Users\noahs\Desktop\New folder\folder\lib\site-packages\open_clip\loss.py", line 2, in <module>
    import torch.distributed.nn
  File "C:\Users\noahs\Desktop\New folder\folder\lib\site-packages\torch\distributed\nn\__init__.py", line 1, in <module>
    from .api.remote_module import RemoteModule
  File "C:\Users\noahs\Desktop\New folder\folder\lib\site-packages\torch\distributed\nn\api\remote_module.py", line 25, in <module>
    from torch.distributed.rpc.internal import _internal_rpc_pickler
  File "C:\Users\noahs\Desktop\New folder\folder\lib\site-packages\torch\distributed\rpc\internal.py", line 12, in <module>
    from torch._C._distributed_rpc import _get_current_rpc_agent
ModuleNotFoundError: No module named 'torch._C._distributed_rpc'; 'torch._C' is not a package

heres the code im running:

import open_clip

model, _, preprocess = open_clip.create_model_and_transforms('ViT-B-32-quickgelu', pretrained='laion400m_e32')

Not sure how to handle this. Ive installed torch multiple times on different versions and dont know why I cant use the library.

clip modulenotfounderror python
2 Answers

OpenClip is a separate module. You must install it separately.

pip install open_clip_torch

After that things should work as one expects.

user_19938370

You need to run the setup.py file:

This is the order of installation:

  1. Install with pip
pip3 install open_clip_torch
  1. Find folder of package (python script)
# Import and print the file. The output will be the file's location
# Go to the modules main folder
import open_clip
print(open_clip.__file__)

Navigate to the module's folder

  1. Find setup.py. And run.
python3 setup.py install

Docs

Edit

See this link

From ModuleNotFoundError: No module named 'torch._C'

user_141412230
Related