diff --git a/Dockerfile b/Dockerfile
index 3b98d71..c9679d6 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,14 +1,18 @@
-FROM sealos.hub:5000/nvidia/cuda:11.4.3-devel-ubuntu20.04
+#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 python3 python3-pip vim git openssh-server vim iproute2 iputils-ping &&  rm -rf /var/lib/apt/lists/*
+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  jupyterlab-language-pack-zh-CN
+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/lib/python3.8/dist-packages/notebook/static/bundle.js
+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
diff --git a/start.sh b/start.sh
index 1915553..f3008ed 100755
--- a/start.sh
+++ b/start.sh
@@ -1,3 +1,3 @@
 #!/bin/bash
 service ssh start
-jupyter-notebook --NotebookApp.token=$JUPYTER_PW   --NotebookApp.disable_check_xsrf=True --debug --allow-root --ip=0.0.0.0 --notebook-dir=/
+jupyter-lab --NotebookApp.token=$JUPYTER_PW   --NotebookApp.disable_check_xsrf=True --debug --allow-root --ip=0.0.0.0 --notebook-dir=/