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): ...@@ -178,14 +178,11 @@ Install TE (optionally specifying the framework):
.. code-block:: bash .. code-block:: bash
git clone https://github.com/NVIDIA/TransformerEngine.git
cd TransformerEngine
# Execute one of the following command # Execute one of the following command
NVTE_FRAMEWORK=all pip install . # Build TE for all supported frameworks. NVTE_FRAMEWORK=all pip install git+https://github.com/NVIDIA/TransformerEngine.git@main # Build TE for all supported frameworks.
NVTE_FRAMEWORK=pytorch pip install . # Build TE for PyTorch only. NVTE_FRAMEWORK=pytorch pip install git+https://github.com/NVIDIA/TransformerEngine.git@main # Build TE for PyTorch only.
NVTE_FRAMEWORK=jax pip install . # Build TE for JAX only. NVTE_FRAMEWORK=jax pip install git+https://github.com/NVIDIA/TransformerEngine.git@main # Build TE for JAX only.
NVTE_FRAMEWORK=tensorflow pip install . # Build TE for TensorFlow 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. 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