enginex-ascend-910-diffusers/Dockerfile.a100

FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel
WORKDIR /workspace
# ENV PT_SDPA_ENABLE_HEAD_DIM_PADDING=1
RUN pip install diffusers==0.34.0 transformers sentencepiece
COPY main.py test.sh dataset.json /workspace/
update base image of a100 2025-08-26 17:31:39 +08:00			`FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel`
add Dockerfile for A100 2025-08-21 11:00:53 +08:00			`WORKDIR /workspace`
			`# ENV PT_SDPA_ENABLE_HEAD_DIM_PADDING=1`
fix diffusers version 2025-08-28 11:17:44 +08:00			`RUN pip install diffusers==0.34.0 transformers sentencepiece`
add Dockerfile for A100 2025-08-21 11:00:53 +08:00			`COPY main.py test.sh dataset.json /workspace/`