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

# Crée un user HF compatible
RUN useradd -m -u 1000 user

# Dépendances système requises
RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
    python3.11 python3.11-dev python3-pip python3-venv \
    build-essential curl git git-lfs wget procps bash \
    openssh-client unzip nano vim htop make cmake \
    && rm -rf /var/lib/apt/lists/*

# Alias python
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.11 1

# Rust
RUN curl https://sh.rustup.rs -sSf | bash -s -- -y
ENV PATH="/root/.cargo/bin:${PATH}"

# Protoc
RUN PROTOC_ZIP=protoc-21.12-linux-x86_64.zip && \
    curl -OL https://github.com/protocolbuffers/protobuf/releases/download/v21.12/$PROTOC_ZIP && \
    unzip -o $PROTOC_ZIP -d /usr/local bin/protoc && \
    unzip -o $PROTOC_ZIP -d /usr/local 'include/*' && \
    rm -f $PROTOC_ZIP

CMD ["tail", "-f", "/dev/null"]