From 104a539a22ebbfda006149a15cb44476df7be68e Mon Sep 17 00:00:00 2001 From: SolitaryThinker Date: Wed, 24 Dec 2025 09:32:42 +0000 Subject: [PATCH] update docker --- docker/Dockerfile.python3.11 | 11 ++--------- docker/Dockerfile.python3.12 | 11 ++--------- docker/Dockerfile.python3.12.cuda12.9.1 | 11 ++--------- docker/rocm7_1.Dockerfile | 6 +++--- 4 files changed, 9 insertions(+), 30 deletions(-) diff --git a/docker/Dockerfile.python3.11 b/docker/Dockerfile.python3.11 index 769b4a1fa..83da01cee 100644 --- a/docker/Dockerfile.python3.11 +++ b/docker/Dockerfile.python3.11 @@ -55,17 +55,10 @@ RUN source $HOME/.local/bin/env && \ echo 'source /opt/venv/bin/activate' >> /root/.bashrc && \ echo 'if [ -n "$ZSH_VERSION" ] && [ -f ~/.zshrc ]; then . ~/.zshrc; elif [ -f ~/.bashrc ]; then . ~/.bashrc; fi' > /root/.profile -# Install STA (Sliding Tile Attention) +# Install FastVideo Kernels RUN source $HOME/.local/bin/env && \ source /opt/venv/bin/activate && \ - cd csrc/attn/sliding_tile_attn && \ - git submodule update --init --recursive && \ - python setup.py install - -# Install VSA -RUN source $HOME/.local/bin/env && \ - source /opt/venv/bin/activate && \ - cd csrc/attn/video_sparse_attn && \ + cd csrc/fastvideo_kernel && \ git submodule update --init --recursive && \ python setup.py install diff --git a/docker/Dockerfile.python3.12 b/docker/Dockerfile.python3.12 index f1ebe560a..106320fa0 100644 --- a/docker/Dockerfile.python3.12 +++ b/docker/Dockerfile.python3.12 @@ -55,17 +55,10 @@ RUN source $HOME/.local/bin/env && \ echo 'source /opt/venv/bin/activate' >> /root/.bashrc && \ echo 'if [ -n "$ZSH_VERSION" ] && [ -f ~/.zshrc ]; then . ~/.zshrc; elif [ -f ~/.bashrc ]; then . ~/.bashrc; fi' > /root/.profile -# Install STA (Sliding Tile Attention) +# Install FastVideo Kernels RUN source $HOME/.local/bin/env && \ source /opt/venv/bin/activate && \ - cd csrc/attn/sliding_tile_attn && \ - git submodule update --init --recursive && \ - python setup.py install - -# Install VSA -RUN source $HOME/.local/bin/env && \ - source /opt/venv/bin/activate && \ - cd csrc/attn/video_sparse_attn && \ + cd csrc/fastvideo_kernel && \ git submodule update --init --recursive && \ python setup.py install diff --git a/docker/Dockerfile.python3.12.cuda12.9.1 b/docker/Dockerfile.python3.12.cuda12.9.1 index 5ecaf95d0..b858f631d 100644 --- a/docker/Dockerfile.python3.12.cuda12.9.1 +++ b/docker/Dockerfile.python3.12.cuda12.9.1 @@ -55,17 +55,10 @@ RUN source $HOME/.local/bin/env && \ echo 'source /opt/venv/bin/activate' >> /root/.bashrc && \ echo 'if [ -n "$ZSH_VERSION" ] && [ -f ~/.zshrc ]; then . ~/.zshrc; elif [ -f ~/.bashrc ]; then . ~/.bashrc; fi' > /root/.profile -# Install STA (Sliding Tile Attention) +# Install FastVideo Kernels RUN source $HOME/.local/bin/env && \ source /opt/venv/bin/activate && \ - cd csrc/attn/sliding_tile_attn && \ - git submodule update --init --recursive && \ - python setup.py install - -# Install VSA -RUN source $HOME/.local/bin/env && \ - source /opt/venv/bin/activate && \ - cd csrc/attn/video_sparse_attn && \ + cd csrc/fastvideo_kernel && \ git submodule update --init --recursive && \ python setup.py install diff --git a/docker/rocm7_1.Dockerfile b/docker/rocm7_1.Dockerfile index 96cfc5cd2..9f286de83 100644 --- a/docker/rocm7_1.Dockerfile +++ b/docker/rocm7_1.Dockerfile @@ -48,11 +48,11 @@ RUN source $HOME/.local/bin/env && \ echo 'source /opt/venv/bin/activate' >> /root/.bashrc && \ echo 'if [ -n "$ZSH_VERSION" ] && [ -f ~/.zshrc ]; then . ~/.zshrc; elif [ -f ~/.bashrc ]; then . ~/.bashrc; fi' > /root/.profile -# Install STA (Sliding Tile Attention) +# Install FastVideo Kernels RUN source $HOME/.local/bin/env && \ source /opt/venv/bin/activate && \ - cd csrc/attn/sliding_tile_attn && \ + cd csrc/fastvideo_kernel && \ git submodule update --init --recursive && \ - SKIP_SM90_EXT=1 pip install --no-build-isolation -e . + python setup.py install EXPOSE 22