EngineX-Ascend/enginex-ascend-910-diffusers
10
0
Fork 1
Code Issues Pull Requests Actions Projects Releases Wiki Activity

update base image of a100

Browse Source
This commit is contained in:
Zhang Hao
2025-08-26 17:31:39 +08:00
parent b9906fa791
commit fcf7f30797
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
Dockerfile.a100
View File

@@ -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/