Spaces:
Build error
Build error
File size: 1,029 Bytes
2df1810 65555f5 11f2c82 5b63bfa 52d2f6a 11f2c82 65555f5 5b63bfa 11f2c82 5b63bfa 65555f5 11f2c82 65555f5 11f2c82 5b63bfa 52d2f6a 65555f5 2df1810 11f2c82 2df1810 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
FROM python:3.8-slim
# Set environment variables
ENV PYTHONUNBUFFERED=1 \
DEBIAN_FRONTEND=noninteractive \
PIP_NO_CACHE_DIR=1
# Install system dependencies
RUN apt-get update && apt-get install -y --no-install-recommends \
build-essential \
libopenblas-dev \
git \
curl \
cmake \
g++ \
make \
&& rm -rf /var/lib/apt/lists/*
# Install Python packages in optimal order
COPY requirements.txt .
RUN pip install --upgrade pip setuptools wheel && \
pip install torch==2.2.1 --index-url https://download.pytorch.org/whl/cpu && \
pip install ninja==1.11.1 && \
pip install transformers==4.38.2 && \
pip install accelerate==0.27.2 && \
pip install deepspeed==0.13.1 && \
pip install datasets==2.18.0 && \
pip install huggingface_hub==0.20.3 && \
pip install unsloth==0.2.0 && \
pip install vllm==0.3.0 --no-deps && \
pip install fastapi==0.109.1 uvicorn==0.27.0
WORKDIR /app
COPY . /app
RUN mkdir -p /app/checkpoints /app/logs
CMD ["python", "train.py"] |