【发布时间】:2021-05-20 20:40:07
【问题描述】:
我使用 pytorch 运行了一个 ML 模型并使用 flask 创建了一个应用程序。在我的本地机器和服务器上一切正常。新学的 docker 镜像,希望把所有东西都打包进去。我有 mac 并安装了 docker 桌面。经过多次尝试和配置,基于我创建的图像的容器在完成运行 py 代码后停止。
设置 dockerfile 的代码
FROM python:3.7.9-stretch
RUN apt-get update
WORKDIR /usr/src/App
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
COPY . .
RUN ls -la .
CMD ["python3", "app.py", "--host=0.0.0.0", "--port=8000"]
构建图像的代码:
docker build -t app:v1 .
运行容器的代码:
docker run -p 8000:8000 app:v1
在我的 py 文件中,我还指定了:
if __name__ == "__main__":
app.run(host='0.0.0.0',port=8000, debug=True)
代码运行完毕后,容器退出。
我查看了日志文件,似乎有一些错误,但我认为这不会导致任何事情,因为它在错误之后还说如果我使用 gpu,那么我可以忽略它(我没有使用 gpu)
【问题讨论】:
标签: docker tensorflow dockerfile huggingface-transformers