FROM docker.io/nvidia/cuda:11.8.0-devel-ubuntu22.04

ARG APT_MIRROR=https://mirrors.bfsu.edu.cn
ARG ROOT_PASSWD=miao

RUN \
    sed -i "s#http://archive.ubuntu.com#${APT_MIRROR}#g" /etc/apt/sources.list && \
    sed -i "s#http://security.ubuntu.com#${APT_MIRROR}#g" /etc/apt/sources.list && \
    \
    echo root:${ROOT_PASSWD} | chpasswd && \
    \
    apt update && \
    apt install --yes ssh tini && \
    apt clean && \
    rm -rf /var/lib/apt/lists/* && \
    \
    mkdir /run/sshd && \
    sed -i 's/#PermitRootLogin.*$/PermitRootLogin yes/' /etc/ssh/sshd_config && \
    sed -i 's#SSHD_OPTS=#SSHD_OPTS=""#' /etc/default/ssh

ENTRYPOINT [ "/usr/bin/tini", "--" ]
CMD [ "/usr/sbin/sshd", "-D" ]