| FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-devel | |
| ENV PYTHONUNBUFFERED 1 | |
| WORKDIR /usr/src/app | |
| # Install packages | |
| RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/* | |
| COPY requirements.txt ./ | |
| RUN pip install --no-cache-dir -r requirements.txt | |
| COPY . . | |