axlearn
axlearn copied to clipboard
Generate requirement files and pyproject.toml for an AXLearn GPU workflow using seed-env
Steps:
- 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.
- pip install seed-env
- run
seed-env --local-requirements=gpu_requirements.txt --host-name=axlearn --seed-commit=779c6aefcac3ff0149b28e183770cc1da0879c84 --hardware=gpu - Find the generated lock files and pyproject.toml within the generated_env/ folder.
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.