sunjichen/enginex-ascend-910-diffusers
1
0
Fork 0
forked from EngineX-Ascend/enginex-ascend-910-diffusers
Code Pull Requests Activity

add Dockerfile for A100

Browse Source
This commit is contained in:
ZHANG Hao
2025-08-21 11:00:53 +08:00
parent 9009b27d65
commit ef778f63e0
2 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
Dockerfile.a100 Normal file
View File

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

3
run_in_docker_a100.sh Executable file
View File

@@ -0,0 +1,3 @@
#! /usr/bin/env bash
image=harbor-contest.4pd.io/zhanghao/diffusers:a100-0.2
docker run -it -v /home/zhanghao/workspace:/workspace -v /mnt:/mnt $image bash