FROM pytorch/pytorch:2.12.1-cuda13.0-cudnn9-devel
RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y -o Dpkg::Options::="--force-confnew" --no-install-recommends \
        vim curl git unzip zip wget openssh-server iputils-ping locales tzdata \
	&& sed -i "s/.*PermitRootLogin.*/PermitRootLogin yes/g" /etc/ssh/sshd_config \
	&& apt-get clean && rm -rf /var/lib/apt/lists/* \
	&& mkdir -p /etc/jupyter /ai

COPY run.sh /usr/local/bin/
COPY notebook /usr/local/bin/
COPY jupyter_notebook_config.py /etc/jupyter/

RUN chmod +x /usr/local/bin/run.sh /usr/local/bin/notebook \
	&& ln -sf /usr/share/zoneinfo/Asia/Shanghai /etc/localtime \
	&& pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple \
	&& pip install --no-cache-dir --break-system-packages notebook tensorboard

EXPOSE 22 8888

ENV LANG=C.UTF-8
ENV SHELL=/bin/bash
WORKDIR /ai

CMD ["run.sh"]
