FROM pytorch/pytorch:2.7.1-cuda12.8-cudnn9-runtime
ENV PIP_DISABLE_PIP_VERSION_CHECK=1 OMP_NUM_THREADS=2 MKL_NUM_THREADS=2 PYTHONDONTWRITEBYTECODE=1
RUN python -m pip install --no-cache-dir numpy==1.26.4 scipy==1.13.1 opencv-python-headless==4.10.0.84 mmengine==0.10.7 mmcv-lite==2.1.0 einops==0.8.1 decord==0.6.0 matplotlib==3.9.4 prettytable==3.16.0 pyyaml==6.0.2 tqdm==4.67.1 thop==0.1.1.post2209072238 pandas==2.2.3
COPY sources/mmaction2-a5a167dff2399e2d182a60332325f9c0d4663517 /opt/mmaction2/
COPY sources/YOWOv2-9e8d23c11ad26ef5e6cfc01e3f4f1112dee634bf /opt/yowov2/
RUN python -m pip install --no-cache-dir --no-deps /opt/mmaction2
RUN python -m pip uninstall -y opencv-python opencv-python-headless
RUN python -m pip install --no-cache-dir --no-deps opencv-python-headless==4.10.0.84
RUN python -m pip install --no-cache-dir importlib-metadata==8.7.0
ENV PYTHONPATH=/opt/mmaction2
WORKDIR /work
CMD ["python", "--version"]