Spaces:
Runtime error
Runtime error
File size: 632 Bytes
d7a7846 491fb88 ceee986 d7a7846 |
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 |
FROM runpod/pytorch:2.1.0-py3.10-cuda11.8.0
WORKDIR /workspace
# Install system dependencies
RUN apt-get update && apt-get install -y \
libgl1-mesa-glx \
libglib2.0-0 \
&& rm -rf /var/lib/apt/lists/*
# Copy application code
COPY . .
# Install requirements first
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
# Install the package in editable mode
RUN pip install --no-cache-dir -e .
# Create directories for models and temporary files
RUN mkdir -p /model-weights /workspace/temp
# Download models (if needed)
# RUN python download_models.py
EXPOSE 8000
CMD ["python", "handler.py"]
|