Skip to content

Add requirements_nvidia.txt, Install flagtree by default - #1299

Closed
wenxian1 wants to merge 1 commit into
flagos-ai:masterfrom
wenxian1:add_requirements_nvidia_txt
Closed

Add requirements_nvidia.txt, Install flagtree by default#1299
wenxian1 wants to merge 1 commit into
flagos-ai:masterfrom
wenxian1:add_requirements_nvidia_txt

Conversation

@wenxian1

Copy link
Copy Markdown
Contributor

PR Category

Feature

Type of Change

New Feature

Description

Add the requirements_nvidia.txt file to support the default installation of flagtree on machines with an NVIDIA backend.
Add the flagos private source to requirements_nvidia.txt to obtain the flagtree wheel package.

Issue

Progress

  • Change is properly reviewed (1 reviewer required, 2 recommended).
  • Change is responded to an issue.
  • Change is fully covered by a UT.

Performance

After installing dependencies using requirements_nvidia.txt, the pytest for flagtree and flaggems both pass.
img_v3_02tb_db2497fd-0068-42a7-94fe-e4d9a0b6c0cg

Comment thread requirements_nvidia.txt
--extra-index-url https://pypi.tuna.tsinghua.edu.cn/simple
--trusted-host resource.flagos.net pypi.tuna.tsinghua.edu.cn
torch>=2.2.0
packaging

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Please specify the version range as you have done for torch.
Pinning a version (or a version range) helps make sure the build result is reproducible.

Suggested change
packaging
packaging>=25.0

@tengqm

tengqm commented Dec 29, 2025

Copy link
Copy Markdown
Contributor

@wenxian1 How are we gonna use this new requirements file? In which stage? Can we incorporate this list of requirments into the pyproject.toml file?

@wenxian1 wenxian1 closed this Dec 29, 2025
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.

2 participants