Skip to content

[dockerfile] temp reinstall triton==3.6.0 for accuracy#726

Merged
valarLip merged 5 commits into
mainfrom
guanbao/update_triton_version
May 10, 2026
Merged

[dockerfile] temp reinstall triton==3.6.0 for accuracy#726
valarLip merged 5 commits into
mainfrom
guanbao/update_triton_version

Conversation

@gbyu-amd
Copy link
Copy Markdown
Contributor

@gbyu-amd gbyu-amd commented May 9, 2026

Motivation

Technical Details

Test Plan

Test Result

Submission Checklist

Comment thread docker/Dockerfile Outdated
rm -f /tmp/triton_pip_show.txt && \
"${VENV_PYTHON}" -c "import triton; print(f'triton.__version__ = {triton.__version__}')"

# unisntall existing triton (acc issue) and install known good version
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

typo uninstall?

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Good catch. Updated the comment.

Comment thread docker/Dockerfile
RUN pip uninstall -y triton && \
pip install triton==3.6.0
RUN pip show triton || true

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

可以加一些注释,解释为什么放弃前面安装的所有triton版本,选择直截了当安装3.6.0

或者是不是修改兜底建构逻辑 [OOT] Restore base image triton
atom-vllm最后时刻出docker的时候,会restore triton版本,保持和atom docker一致,现在atom-vllm要pin住3.6.0,是不是直接在兜底逻辑里面pin住比较好

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Updated comment for background.

@valarLip valarLip merged commit 3e06687 into main May 10, 2026
31 of 48 checks passed
@valarLip valarLip deleted the guanbao/update_triton_version branch May 10, 2026 02:07
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants