python - 导入 open_clip 时出现 ModuleNotFoundError

Question

导入 open_clip 时出错，导入 open_clip 时，我总是会收到此错误:

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

这是我正在运行的代码:

import open_clip

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

不确定如何处理。我在不同版本上多次安装了 torch，但不知道为什么我不能使用该库。

Answer 1

OpenClip是一个单独的模块。您必须单独安装它。

pip install open_clip_torch

在那之后，事情应该会按预期进行。