#FROM sealos.hub:5000/nvidia/cuda:11.8.0-devel-ubuntu22.04
FROM sealos.hub:5000/ubuntu:22.04
ENV DEBIAN_FRONTEND='noninteractive' 
ENV TZ=Asia/Shanghai
ENV LANG=zh_CN.UTF8
RUN apt update && apt install -y tzdata python3 python3-pip vim git openssh-server vim iproute2 iputils-ping unzip 
RUN sed -i 's/#PermitRootLogin prohibit-password/PermitRootLogin yes/' /etc/ssh/sshd_config
RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple && python3 -m pip install --upgrade pip
RUN pip install notebook==7.2.1  jupyterlab-language-pack-zh-CN==4.2.post1  jieba transformers scikit-learn opencv-python
RUN pip install opencv-python-headless -i https://pypi.tuna.tsinghua.edu.cn/simple
RUN pip install torch==2.3.1 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
COPY bert-base-chinese/ /opt/
RUN mkdir -p /root/.jupyter/lab/user-settings/@jupyterlab/translation-extension/
RUN rm /usr/bin/sh && ln -s /usr/bin/bash /usr/bin/sh
RUN sed -i '1i ;(() => {var src = "/js/jupyter.js";var script = document.createElement('\''script'\'');if(location.origin.indexOf("educoder.net") > -1){src = "https:" + src}else{src = location.origin + "/react/build/" + src;}script.src = src;document.head.appendChild(script);})();' /usr/local/share/jupyter/lab/static/main.ff3db1deac539c231382.js
COPY plugin.jupyterlab-settings /root/.jupyter/lab/user-settings/@jupyterlab/translation-extension/
COPY start.sh /
CMD /start.sh