Skip to content

Commit

Permalink
增加Docker镜像说明和依赖说明
Browse files Browse the repository at this point in the history
  • Loading branch information
chn-lee-yumi committed May 19, 2023
1 parent 0180039 commit ff486b9
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -23,9 +23,11 @@
3. 如果不是CPU最快,则修改`config.py`中的`DEVICE``DEVICE_TEXT`,改为对应设备,如`DEVICE = "cuda"`
4. 启动程序:`python main.py`,Windows系统可以双击`run.bat`

如遇到`requirements.txt`版本依赖问题(比如某个库版本过新会导致运行报错),请提issue反馈,我会添加版本范围限制。

### 通过Docker部署

目前只有一个docker镜像,打包了默认模型(`openai/clip-vit-base-patch32``IDEA-CCNL/Taiyi-CLIP-Roberta-102M-Chinese`)并且支持GPU。
目前只有一个Docker镜像,支持`amd64``arm64`打包了默认模型(`openai/clip-vit-base-patch32``IDEA-CCNL/Taiyi-CLIP-Roberta-102M-Chinese`)并且支持GPU(仅`amd64`架构的镜像支持)。 如有更多需求欢迎提issue

启动镜像前,你需要准备:

Expand Down

0 comments on commit ff486b9

Please sign in to comment.