add Dockerfile for A100

2025-08-21 11:00:53 +08:00
parent 9009b27d65
commit ef778f63e0
2 changed files with 8 additions and 0 deletions
--- a/Dockerfile.a100
+++ b/Dockerfile.a100
@@ -0,0 +1,5 @@
+FROM harbor-contest.4pd.io/zhangyiqun/public/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
+COPY main.py test.sh dataset.json /workspace/