尧图网站设计 尧图网站设计YAOTU DESIGN
ARTICLE DETAIL

资讯详情

深耕网站设计与一线实操的经验洞察。

Cog CLI 完整实操指南:5 分钟把机器学习模型容器化部署上线

Cog CLI 完整实操指南:5 分钟把机器学习模型容器化部署上线 Cog CLI 完整实操指南5 分钟把机器学习模型容器化部署上线【免费下载链接】cogContainers for machine learning项目地址: https://gitcode.com/GitHub_Trending/co/cogCog CLI是容器化机器学习模型部署的命令行工具你只需用 Python 写模型代码它负责装依赖、建环境、打包镜像、起服务。跟着走一遍5 分钟就能完成从 init 到 push 的全流程不用自己手写 Docker 镜像。新项目怎么起步一条命令 init它解决的问题手写第一份cog.yaml容易把字段写错。cog init会在当前目录生成三个文件cog.yaml环境配置Python 版本、依赖文件、是否需要 GPU、run.py模型入口一个带setup()和run()的Runner类骨架、requirements.txt。模板直接嵌在二进制里断网也能用。⚠️ 这里有个坑它不会覆盖已存在的文件目录里已有run.py时会提示Skipped existing直接跳过——想重来就先换个干净目录。本地第一次预测怎么跑cog run它解决的问题不想在宿主机装 torch 和一堆依赖。模型代码全部跑在容器里你机器上只要有 Docker 就行。CLI 一条链路搞定解析输入 → 自动构建镜像 → 起容器 → 发一次 HTTP 预测请求 → 停容器。cog run -i imagekodim24.png -i blur8前缀表示值是个本地文件CLI 会读文件再传给容器不加会被当成普通字符串。输出按模型声明的类型呈现字符串直接打印文件输出自动写到本地磁盘数字和对象输出格式化 JSON。上面这条跑的就是仓库里 examples/blur/ 的图像模糊模型处理完长这样 也可以直接对已构建好的镜像跑预测把镜像引用作为第一个参数传进去跳过本地构建拉下来就能用⚠️ 90% 的人第一次会在这里卡住输入校验发生在构建之前-i blurabc这种类型错误会当场报错、容器根本没启动。先别慌这是故意的——省掉你等几分钟构建的冤枉时间。调试容器环境cog exec 和 cog servecog exec先查环境再查代码它解决的问题模型起不来第一步是确认容器里依赖装没装齐、CUDA 能不能用。cog exec python -c import torch; print(torch.cuda.is_available())当前目录会以卷的形式挂到容器内的/src工作目录也设在那里感觉就像在项目目录里直接敲命令。要跑 Web 服务就加-p 8888把端口发布出来。cog serve用和生产一样的 HTTP 接口调模型它解决的问题上线前想验证真实调用链路。cog serve默认在宿主机 8393 端口监听起来之后POST /predictions就能发预测/openapi.json看接口规范/health-check看健康状态。本地开发时源码是卷挂载进容器的改代码重启即生效不用重新构建。想深挖实现可以看 pkg/cli/serve.go。构建产物cog build 怎么打镜像它解决的问题只想要镜像、不想要预测结果比如推到registry或交给 CI。cog build -t my-model构建时会顺带基于 Python 类型注解静态分析出生成的OpenAPI Schema写进镜像 label——部署方不用起容器就能预校验输入类型。镜像命名优先级-t标签 cog.yaml里的image字段 按项目目录生成的默认名。上线部署cog push 构建并推送它解决的问题把模型从本地一次性推上 registry拿到线上模型地址。cog push r8.im/your-username/my-model两个细节值得知道一是构建开始前会先校验目标引用COG_MODEL环境变量和cog.yaml的model字段冲突会当场报出而不是让你白等几分钟构建二是推送成功后会打印 digest 固定的引用直接复制就能用。⚠️ 第一次 push 大概率失败你没登录。先执行cog login输入 tokenCI 场景用cog login --token-stdin从标准输入传免得卡在交互提示上。新手最容易卡住的 3 个点速查坑现象解法端口混淆curl 5000 没反应5000 是容器内端口宿主机默认监听 8393GPU 没生效日志出现 Missing device drivernvidia 驱动没装CLI 会自动去掉 GPU 重试文件传成字符串模型里收到的是路径文本路径前加-i imagephoto.jpg收尾一句话cog init起步cog run/cog serve/cog exec开发调试cog build出产物cog push上线整条链路 5 条命令模型代码自始至终不在宿主机上执行。环境、依赖、CUDA 版本全部锁在容器镜像里——这就是本地跑通即线上运行。【免费下载链接】cogContainers for machine learning项目地址: https://gitcode.com/GitHub_Trending/co/cog创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
返回列表