FROM nvidia/cuda:11.4.2-cudnn8-devel-ubuntu20.04
RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y -o Dpkg::Options::="--force-confnew" --no-install-recommends \
	python3-dev python3-pip python3-setuptools \
        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/* \
	&& ln -s /usr/bin/python3 /usr/bin/python \
	&& pip install --no-cache-dir -i https://pypi.tuna.tsinghua.edu.cn/simple -U pip \
	&& 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 notebook==6.4.10 torch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 tensorboard==2.12.0 \
        && sed -i "s/DELETE/DELETE', 'OPTIONS/g" /usr/local/lib/python3.8/dist-packages/notebook/terminal/api_handlers.py

EXPOSE 22 8888

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

CMD ["run.sh"]
