Unverified Commit 9627b073 authored by Santosh Bhavani's avatar Santosh Bhavani Committed by GitHub
Browse files

Updated README - Added conda installation (#1826)



* added conda installation
Signed-off-by: default avatarSantosh Bhavani <santosh.bhavani@live.com>

* fix for pypi
Signed-off-by: default avatarKirthi Shankar Sivamani <ksivamani@nvidia.com>

---------
Signed-off-by: default avatarSantosh Bhavani <santosh.bhavani@live.com>
Signed-off-by: default avatarKirthi Shankar Sivamani <ksivamani@nvidia.com>
Co-authored-by: default avatarKirthi Shankar Sivamani <ksivamani@nvidia.com>
parent cd37379d
......@@ -224,6 +224,18 @@ When installing from GitHub, you can explicitly specify frameworks using the env
NVTE_FRAMEWORK=pytorch,jax pip install git+https://github.com/NVIDIA/TransformerEngine.git@stable
conda Installation
^^^^^^^^^^^^^^^^^^
To install the latest stable version with conda from conda-forge:
.. code-block:: bash
# For PyTorch integration
conda install -c conda-forge transformer-engine-torch
# JAX integration (coming soon)
Source Installation
^^^^^^^^^^^^^^^^^^^
......
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