conda环境module ‘clip‘ has no attribute ‘load‘解决

原创

已于 2023-12-12 23:25:50 修改 · 7.4k 阅读

31 ·

CC 4.0 BY-SA版权

文章标签：

#clip #零次学习 #zero-shot #conda

于 2023-11-23 16:34:19 首次发布

文章讲述了用户在尝试使用CLIP模型进行零样本预测时遇到的错误，主要问题是`clip.load()`方法未找到。解决方法包括检查pip包、重新安装相关组件（如torch,torchvision等）以及使用Conda进行环境管理。文章还介绍了Conda的包管理器功能和优势，如环境隔离和依赖性解决。

1 问题描述

运行基于clip的zero-shot预测代码，报错如下：

Traceback (most recent call last):
  File "D:\code\ptcontainer\clip\clipembeding.py", line 38, in <module>
    clip_embeding = ClipEmbeding()
  File "D:\code\ptcontainer\clip\clipembeding.py", line 11, in __init__
    self.model, self.processor = clip.load("d:/models/clip/ViT-L-14-336px.pt", device=self.device)
AttributeError: module 'clip' has no attribute 'load'

执行的代码：

import os
import clip
import torch
from torchvision.datasets import CIFAR100

# Load the model
device = "cuda" if torch.cuda.is_available() else "cpu"
model, preprocess = clip.load('ViT-B/32', device)

# Download the dataset
cifar100 = CIFAR100(root=os.path.expanduser("./data/"), download=True, train=False)

# Prepare the inputs
image, class_id = cifar100[3637]
image_input = preprocess(im