ARG BASE_IMAGE=myelin-node:latest
FROM ${BASE_IMAGE}

ARG TINYGRAD_VERSION=0.12.0
RUN apt-get update && \
    apt-get install -y --no-install-recommends python3-pip && \
    python3 -m pip install \
        --break-system-packages \
        --no-cache-dir \
        "tinygrad==${TINYGRAD_VERSION}" && \
    apt-get purge -y --auto-remove python3-pip && \
    apt-get clean && \
    rm -rf /var/lib/apt/lists/* \
           /var/cache/apt/archives/* \
           /var/cache/apt/*.bin \
           /var/log/apt/* \
           /var/log/dpkg.log \
           /tmp/* \
           /var/tmp/*

ENV DEV=CUDA \
    CUDA_PTX=1