Skip to content

GitLab

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

Document how to install a specific PR. (#488) 


Signed-off-by: default avatarFrederic Bastien <fbastien@nvidia.com>
parent 4d1f92df
Show whitespace changes
Inline Side-by-side
Showing with 7 additions and 0 deletions
docs/installation.rst
View file @ 7a025c6b
...@@ -65,6 +65,13 @@ Execute the following command to install the latest development build of Transfo ...@@ -65,6 +65,13 @@ Execute the following command to install the latest development build of Transfo
This will automatically detect if any supported deep learning frameworks are installed and build Transformer Engine support for them. To explicitly specify frameworks, set the environment variable `NVTE_FRAMEWORK` to a comma-separated list (e.g. `NVTE_FRAMEWORK=jax,pytorch`). This will automatically detect if any supported deep learning frameworks are installed and build Transformer Engine support for them. To explicitly specify frameworks, set the environment variable `NVTE_FRAMEWORK` to a comma-separated list (e.g. `NVTE_FRAMEWORK=jax,pytorch`).
In order to install a specific PR, execute after changing NNN to the PR number:
.. code-block:: bash
pip install git+https://github.com/NVIDIA/TransformerEngine.git@refs/pull/NNN/merge
Installation (from source) Installation (from source)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
......
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