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
docker部署训练好的模型(非常简单)
最新推荐文章于 2025-03-13 04:45:00 发布