diff --git a/Dockerfile.a100 b/Dockerfile.a100
index 08f51e3..1a9d182 100644
--- a/Dockerfile.a100
+++ b/Dockerfile.a100
@@ -1,5 +1,5 @@
-FROM harbor-contest.4pd.io/zhangyiqun/public/pytorch:2.6.0-cuda12.4-cudnn9-devel
+FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel
 WORKDIR /workspace
 # ENV PT_SDPA_ENABLE_HEAD_DIM_PADDING=1
-RUN pip install diffusers transformers sentencepiece -i https://nexus.4pd.io/repository/pypi-all/simple
+RUN pip install diffusers transformers sentencepiece
 COPY main.py test.sh dataset.json /workspace/