Spaces:
Running
on
Zero
Running
on
Zero
File size: 443 Bytes
a57579d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
FROM pytorch/pytorch:2.5.1-cuda12.1-cudnn8-runtime
RUN apt-get update && apt-get install -y \
git build-essential python3-dev ninja-build && \
rm -rf /var/lib/apt/lists/*
RUN pip install --upgrade pip
COPY requirements.txt /tmp/requirements.txt
RUN pip install -r /tmp/requirements.txt
RUN pip install git+https://github.com/Dao-AILab/[email protected]#subdirectory=csrc
COPY . /app
WORKDIR /app
CMD ["python", "app.py"] |