Skip to content

Fix/pyt hy video amd dockerfile #149

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
vadseshu wants to merge 3 commits into
base: develop
Choose a base branch
from
Open

Fix/pyt hy video amd dockerfile #149

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
549e9ec
Added gfx950a to pyt_mochi_inference.ubuntu dockerfile to enable fa
vadseshu1 Apr 16, 2026
0c40e04
fix(docker): Hunyuan video AMD image build and runtime imports
vadseshu Apr 20, 2026
33d2d64
Update pyt_hy_video.ubuntu.amd.Dockerfile
vadseshu Jun 24, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
33 changes: 27 additions & 6 deletions docker/pyt_hy_video.ubuntu.amd.Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,10 +6,11 @@ ARG work_dir=/hunyuanvideo

WORKDIR $work_dir

RUN pip install diffusers==0.32.2 distvae yunchang==0.6.0 opencv-python accelerate
#RUN pip install diffusers==0.32.2 distvae yunchang==0.6.0 opencv-python accelerate
RUN pip install transformers==4.56.2 diffusers==0.32.2 distvae yunchang==0.6.0 opencv-python-headless accelerate
RUN pip install imageio imageio-ffmpeg
RUN pip install beautifulsoup4==4.12.3
RUN pip install sentencepiece==0.1.99
RUN pip install sentencepiece>=0.2.0
RUN pip install numpy==1.26.4

# xDiT repository
Expand All @@ -19,7 +20,7 @@ RUN git clone https://github.com/xdit-project/xDiT && \
pip install --no-deps -e .

# flash attn
ARG FA_SHA="22c0358"
ARG FA_SHA="83f9e450cd10e20701fb109db9c7703d376f282b"
ARG FA_REPO="https://github.com/ROCm/flash-attention.git"
ARG PYTORCH_ROCM_ARCH="gfx942"
RUN git clone ${FA_REPO} && \
Expand All @@ -28,12 +29,32 @@ RUN git clone ${FA_REPO} && \
git submodule update --init && \
GPU_ARCHS=${PYTORCH_ROCM_ARCH} python3 setup.py bdist_wheel --dist-dir=dist && \
pip install dist/*.whl;
RUN apt-get update && apt-get install -y \
build-essential \
cmake \
pkg-config \
xxd \
git \
python3-dev \
&& rm -rf /var/lib/apt/lists/*

# RPD profiling
#RUN apt update && \
# apt install -y sqlite3 libsqlite3-dev libfmt-dev
#RUN git clone https://github.com/ROCmSoftwarePlatform/rocmProfileData && \
# cd rocmProfileData && \
# make && make install && \
# cd rocpd_python && python setup.py install && cd .. && \
# cd rpd_tracer && python setup.py install && cd ..


# RPD profiling
RUN apt update && \
apt install -y sqlite3 libsqlite3-dev libfmt-dev
RUN git clone https://github.com/ROCmSoftwarePlatform/rocmProfileData && \
cd rocmProfileData && \
make && make install && \
cd rocpd_python && python setup.py install && cd .. && \
cd rpd_tracer && python setup.py install && cd ..
make && \
make install -C rocpd_python && \
make install -C rpd_tracer && \
ldconfig

2 changes: 1 addition & 1 deletion docker/pyt_mochi_inference.ubuntu.amd.Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ RUN mkdir -p $WORKSPACE_DIR
WORKDIR $WORKSPACE_DIR

ARG FA_REPO="https://github.com/Dao-AILab/flash-attention.git"
ARG PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201
ARG PYTORCH_ROCM_ARCH=gfx950;gfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201
RUN git clone ${FA_REPO}
RUN cd flash-attention \
&& git submodule update --init \
Expand Down