Quantization-Attempts / Dockerfile
Rúben Almeida
Add install torch before performing automq pip install -r requirements.txt
288d5ce
raw
history blame
600 Bytes
FROM python:3.13-slim
# Set the working directory
WORKDIR /app
RUN apt-get update && apt-get install -y build-essential
# Update pip to the latest version
RUN pip install --upgrade pip
RUN pip install -U setuptools wheel
# Install torch
RUN pip install -U torch torchvision torchaudio
# Copy the requirements file into the container
COPY requirements.txt .
# Install the dependencies
RUN pip install --no-cache-dir -r requirements.txt
# Copy the rest of the application code into the container
COPY . .
EXPOSE 7860
ENTRYPOINT [ "fastapi", "run", "main.py", "--host=0.0.0.0", "--port=7860" ]