FROM nvidia/cuda:12.6.0-devel-rockylinux9

RUN dnf -y install \
    python3 \
    python3-devel \
    python3-pip \
    git
