axlearn icon indicating copy to clipboard operation
axlearn copied to clipboard

Generate requirement files and pyproject.toml for an AXLearn GPU workflow using seed-env

Open kanglant opened this issue 6 months ago • 1 comments

Steps:

  1. Copy the necessary GPU container dependencies from AXLearn's pyproject.toml into a requirements.txt file, excluding seqio due to its outdated protobuf upper bound.
  2. pip install seed-env
  3. run seed-env --local-requirements=gpu_requirements.txt --host-name=axlearn --seed-commit=779c6aefcac3ff0149b28e183770cc1da0879c84 --hardware=gpu
  4. Find the generated lock files and pyproject.toml within the generated_env/ folder.

kanglant avatar Jul 28 '25 12:07 kanglant

This pull request has been automatically marked as stale because it has been inactive for 60 days. It will be closed in 7 days if no further activity occurs. If you would like to continue working on this, please remove the stale label or leave a comment.

github-actions[bot] avatar Oct 18 '25 02:10 github-actions[bot]