diff --git a/Dockerfile.a100 b/Dockerfile.a100
new file mode 100644
index 0000000..9284543
--- /dev/null
+++ b/Dockerfile.a100
@@ -0,0 +1,5 @@
+FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel
+
+WORKDIR /workspace
+RUN pip install sentence-transformers -i https://nexus.4pd.io/repository/pypi-all/simple
+COPY . /workspace/
diff --git a/run_in_docker_a100.sh b/run_in_docker_a100.sh
new file mode 100755
index 0000000..c3f8d31
--- /dev/null
+++ b/run_in_docker_a100.sh
@@ -0,0 +1,3 @@
+#! /usr/bin/env bash
+image=harbor-contest.4pd.io/zhanghao/feature:a100-0.1
+docker run -it -v /home/zhanghao/workspace:/workspace -v /mnt:/mnt $image bash
diff --git a/test.sh b/test.sh
new file mode 100755
index 0000000..772b756
--- /dev/null
+++ b/test.sh
@@ -0,0 +1,2 @@
+# python main.py dataset.json output.json /mnt/contest_ceph/zhanghao/models/BAAI/bge-large-zh-v1.5 --device npu
+python main.py dataset.json output.json /mnt/contest_ceph/zhanghao/models/BAAI/bge-large-zh-v1.5 --device cuda