yanzheng / Dockerfile
guge123's picture
Update Dockerfile
553eb16 verified
# 使用官方的 Python Docker 镜像作为基础镜像
FROM python:3.8-slim
# 创建一个非特权用户并设置工作目录
RUN useradd --create-home --shell /bin/bash myuser
WORKDIR /home/myuser/app
# 安装所需的库
RUN pip install --no-cache-dir torch==1.9.0 transformers flask
# 设置TRANSFORMERS_CACHE环境变量
ENV TRANSFORMERS_CACHE /home/myuser/.cache/huggingface
# 复制当前目录内容到容器的 /app 目录下
COPY . .
# 更改文件夹所有者为非特权用户
RUN chown -R myuser:myuser /home/myuser
# 指定默认命令,运行 Python 脚本
CMD ["python", "app.py"]