docker部署训练好的模型(非常简单)
1、安装docker环境(以centos为例):
yum install -y yum-utils # yum工具集
yum-config-manager --add-repo https://mirrors.aliyun.com/docker-ce/linux/centos/docker-ce.repo
yum makecache fast # yum自动检测最快的安装源
yum install -y docker-ce
service docker start
docker version
2、在本地环境下生成依赖文件:
pip freeze > requirements.txt
3、部署相关文件:
制作Dockerfile:
FROM python:3.8
COPY . /predict
WORKDIR /predict
RUN pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
CMD ["python","predict.py"]
FROM python:3.8指定python环境版本,以具体试验为准;
COPY . /predict意为将当前目录的文件复制到容器内部的predict文件夹下;
WORKDIR /predict指定容器的工作目录;
RUN pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple 将环境的依赖拉取;
CMD ["python","predict.py"]指用python运行利用模型预测的脚本。
4、构建镜像(根据当前的目录下的Dockerfile文件构建名为predict的镜像):
docker build -t predict .
5、运行容器:
docker run predict
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐


所有评论(0)