Installation
If you simply want to use this package without modifying it, the best option is to install it as a dependency of your project like you would with any other pip package.
Normal Installation
To install the most recent release version, run
pip install megatron-energon
in your project’s Python environment, which could be a virtualenv, or a conda environment.
You can even install it inside a Dockerfile
to include it in your custom docker container.
Installation for Development
If you want to manage, debug or modify the code of energon itself, we recommend that you clone this repository on your disk.
You can then install the package in editable mode. This way, you can use energon and its CLI scripts while still being able to modify the source code.
First, check out the repository locally:
git clone https://github.com/NVIDIA/Megatron-Energon.git megatron-energon
Then install with your favorite tooling:
Editable installation with uv and just
uv
is a fast modern tool that can replace legacy tools like pip, conda and virtualenv.just
is command runner that simplifies common tasks using thejustfile
we provide.
Check out the official website on how to install uv
.
On this page you can find out how to install just
.
Then, to setup a .venv
and install energon in editable mode:
cd megatron-energon
just dev-sync
The dev-sync
command will setup a local virtual environment in .venv
and install all dependencies.
It will also install energon in editable mode for development inside that venv.
Activate the environment
. .venv/bin/activate
Now you can call the energon
command.
You can also use just
to do a bunch of other things shown below.
Note that you don’t need to activate the venv before running those.
# Run all unit tests
just test
# Run the code linter and format check
just check
# Build the documentation
just docs
# Show all available commands
just help
Editable installation with pip
First make sure you are in some python environment where you want to set up energon. Then install in development mode:
pip install -e ./megatron-energon
Warning
We discourage importing the cloned repo without pip install
You will not be able to use the command line tool
You would have to use hacks to get the package into your
PYTHONPATH
You would need to take care of the dependencies yourself.
Instead, simply install in development mode.