创建用于训练深度学习模型的 docker 镜像

时间:2021-02-19 09:44:00

标签: python docker deep-learning

目前我有一个深度学习模型,我可以用 Anaconda 管理的环境和包来训练它。我正在研究如何将所有依赖项连同我的代码打包到一个 docker 容器中。

我想到的一种方法是执行 pip freeze 以获取我环境中的软件包版本,将它们放入 requirements.txt,然后使用它来运行 pip install -r requirements.txt。这是最好的方法吗?

有人可以向我提供一些入门指南,我对 Docker 不是很熟悉,因此请求答案以提供简单的解释并尽可能参考资源。谢谢。

1 个答案:

答案 0 :(得分:1)

示例 Dockerfile。

FROM python:3.7-alpine
COPY . /app
WORKDIR /app
RUN pip install -r requirements.txt
CMD ["python","file.py"]