FROM nvidia/cuda:12.4.0-devel-ubuntu22.04

RUN apt-get update && apt-get install -y \
    python3 python3-dev python3-venv git curl gcc g++ && \
    ln -sf python3 /usr/bin/python && rm -rf /var/lib/apt/lists/*

# uv 설치
RUN curl -LsSf https://astral.sh/uv/install.sh | sh
ENV PATH="/root/.local/bin:$PATH"

# libcuda stub for triton JIT compilation
ENV LIBRARY_PATH="/usr/local/cuda/lib64/stubs:$LIBRARY_PATH"

COPY requirements.txt /app/requirements.txt
RUN uv pip install --system --no-cache -r /app/requirements.txt

COPY . /app
WORKDIR /app

CMD ["python", "test_multi_lora.py"]