按照源码进行部署,方便接口调用。
环境:python37+cuda10.1+nvdia driver 418.87
1.安装
torch==1.8.1_cu101
torchvision==0.9.1_cu101
tokenizers==0.11.6
transformers==4.20.0
huggingface-hub==0.2.0
timm
ftfy
sentencepiece
open_clip==2.16.0 自行编译一下,在git上下载源码,用python setup.py install --user安装,在requirement中降低pytorch的版本,改成>1.8
opencv-python
pytorch_lightning
更新libstdc++.so.6 1.3.9
更新glibc 2.18
addict
yapf
prettytable
omegaconf==2.1.1
xformers==0.0.2
更改ldm/modules/encoders/modules.py中88行的openai的clip的地址
有几处autocast和pytorch1.8其冲突的几个,全部删掉autocast的推理即可。
ldm/modules/diffusionmodules/util.py", line 126 注掉autocast,