FROM nvidia/cuda:13.0.0-devel-ubuntu24.04

ENV NVIDIA_DRIVER_CAPABILITIES=graphics,utility,compute
ENV DEBIAN_FRONTEND=noninteractive

# System deps
RUN apt-get update && apt-get install -y \
    python3 python3-pip python3-venv \
    build-essential yasm cmake libtool git pkg-config curl \
    libass-dev libfreetype6-dev libvorbis-dev \
    autoconf automake texinfo tmux ffmpeg

COPY . /workspace/

# install_deps.sh copies spark/pyproject.toml to repo root before uv sync
ENV DOCKER_CONTAINER=1
ENV UV_PROJECT_ENVIRONMENT=/opt/gr00t-venv
RUN cd /workspace && bash scripts/deployment/spark/install_deps.sh

# Activate the venv by default and expose nvidia pip package libs (cudnn, cudss, cublas)
ENV VIRTUAL_ENV=/opt/gr00t-venv
ENV PATH="$VIRTUAL_ENV/bin:/usr/local/cuda/bin:$PATH"
ENV TRITON_PTXAS_PATH=/usr/local/cuda/bin/ptxas
ENV CUDA_HOME=/usr/local/cuda-13.0
ENV CUDA_PATH=/usr/local/cuda-13.0
ENV CPATH="/usr/local/cuda-13.0/include:${CPATH:-}"
ENV C_INCLUDE_PATH="/usr/local/cuda-13.0/include:${C_INCLUDE_PATH:-}"
ENV CPLUS_INCLUDE_PATH="/usr/local/cuda-13.0/include:${CPLUS_INCLUDE_PATH:-}"
ENV LD_LIBRARY_PATH="$VIRTUAL_ENV/lib/python3.12/site-packages/torch/lib:$VIRTUAL_ENV/lib/python3.12/site-packages/nvidia/cu13/lib:$VIRTUAL_ENV/lib/python3.12/site-packages/nvidia/cudss/lib:$VIRTUAL_ENV/lib/python3.12/site-packages/nvidia/cudnn/lib:${LD_LIBRARY_PATH:-}"

WORKDIR /workspace/gr00t