使用RTX3090部署CLIP流程
参考网址:clip代码安装实操
运行环境
PyTorch 1.8.1
Python 3.8(ubuntu18.04)
Cuda 11.1
GPU RTX 3090(24GB) * 1
CPU 14 vCPU Intel Xeon Processor (Icelake)
配置流程
首先,点击CLIP的官网github网址。
并跟着官网的流程进行配置。
首先创建一个conda
conda create --name clip python=3.7
conda init bash && source /root/.bashrc
conda activate clip
之后再运行以下代码:
pip install ftfy regex tqdm
再运行如下代码:
运行如下代码的时候
pip install git+https://github.com/openai/CLIP.git
运行这两行代码的时候pytorch会自动安装。这里自动安装了torch1.13.1-cp37
成功安装好后,我们运行命令:
conda list
这时候,看里面的内容显示,我们已经安装上了CLIP库了,这时候我们就可以用了。
如不能在线直接安装成功的,得需要去github上下载压缩包本地安装:
上传解压后,得到如下文件:
进入这个CLIP目录,运行以下安装代码,可以本地安装成功:
pip install -r requirements.txt
python setup.py install
运行demo :
在CLIP目录下创建一个文件test.py:
import torch
import clip
from PIL import Image
device = "cuda" if torch.cuda.is_available