Skip to content

GitLab

Unverified Commit 7bb2af35 authored by Frédéric Bastien's avatar Frédéric Bastien Committed by GitHub
Browse files

Simplify dev installation guide. (#157) 


Signed-off-by: default avatarFrederic Bastien <fbastien@nvidia.com>
parent d3d7ed2c
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 7 deletions
README.rst
View file @ 7bb2af35
......@@ -178,14 +178,11 @@ Install TE (optionally specifying the framework):
.. code-block:: bash
git clone https://github.com/NVIDIA/TransformerEngine.git
cd TransformerEngine
# Execute one of the following command
NVTE_FRAMEWORK=all pip install . # Build TE for all supported frameworks.
NVTE_FRAMEWORK=pytorch pip install . # Build TE for PyTorch only.
NVTE_FRAMEWORK=jax pip install . # Build TE for JAX only.
NVTE_FRAMEWORK=tensorflow pip install . # Build TE for TensorFlow only.
NVTE_FRAMEWORK=all pip install git+https://github.com/NVIDIA/TransformerEngine.git@main # Build TE for all supported frameworks.
NVTE_FRAMEWORK=pytorch pip install git+https://github.com/NVIDIA/TransformerEngine.git@main # Build TE for PyTorch only.
NVTE_FRAMEWORK=jax pip install git+https://github.com/NVIDIA/TransformerEngine.git@main # Build TE for JAX only.
NVTE_FRAMEWORK=tensorflow pip install git+https://github.com/NVIDIA/TransformerEngine.git@main # Build TE for TensorFlow only.
If the framework is not explicitly specified, TE will be built for PyTorch only.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment