FROM pytorch-gpu:2.5.1-cu12.3-py310

COPY git-clone.sh /opt

RUN chmod +x /opt/git-clone.sh \
  && pip uninstall -y opencv eetq

COPY shenergy/ /pkgs

RUN pip install --no-deps /pkgs/* \
  && rm -rf /pkgs
