FROM nvidia/cuda:12.0.0-cudnn8-devel-ubuntu22.04
ARG DEBIAN_FRONTEND=noninteractive

ENV PYTHONUNBUFFERED=1

RUN apt-get update && apt-get install --no-install-recommends -y \
  build-essential \
  python3.11 \
  python3-pip \
  git \
  ffmpeg \
  && apt-get clean && rm -rf /var/lib/apt/lists/*
WORKDIR /workspace

COPY scripts/entrypoint.sh /workspace/entrypoint.sh
RUN chmod +x /workspace/entrypoint.sh
RUN which python3
RUN free -m
RUN which nvidia-smi
RUN nvidia-smi

CMD ["/workspace/entrypoint.sh"]