Skip to content

Commit

Permalink
updating README (#2791)
Browse files Browse the repository at this point in the history
quick update to reflect packages updates.
  • Loading branch information
jjsjann123 authored Aug 14, 2024
1 parent 816c3d6 commit 6f9d811
Showing 1 changed file with 8 additions and 7 deletions.
15 changes: 8 additions & 7 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,14 +10,15 @@ A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")

## Installation

We publish nightly wheel packages on https://pypi.nvidia.com
We publish nightly wheel packages on https://pypi.nvidia.com, while build against stable torch version on https://pypi.org.

built-env | cuda 11.8 | cuda 12.1
:---: | :---: | :---:
torch 2.2 | nvfuser-cu118-torch22 | nvfuser-cu121-torch22
torch nightly wheel | nvfuser-cu118 | nvfuser-cu121
built-env | cuda 12.1
:---: | :---:
torch 2.3 | nvfuser-cu121-torch23
torch 2.4 | nvfuser-cu121-torch24
torch nightly wheel | nvfuser-cu121

Note that nvfuser built against torch-2.2 isn't compatible with nightly pytorch wheel, so ensure you pick the right version suiting your environment.
Note that nvfuser built against stable torch version isn't compatible with nightly pytorch wheel, so ensure you pick the right version suiting your environment.

### nightly nvfuser pip wheel

Expand All @@ -29,7 +30,7 @@ Note that this may uninstall your local pytorch installation and install the com

### nvfuser pip wheel against pytorch stable release

nvfuser pip wheel built against stable torch releases is published on pypi.org. Pick the right cuda toolkit version to match your torch installation. e.g. `pip install nvfuser-cu121-torch22`
nvfuser pip wheel built against stable torch releases is published on pypi.org. Pick the right cuda toolkit version to match your torch installation. e.g. `pip install nvfuser-cu121-torch24`

PyPI: [https://pypi.org/project/nvfuser/](https://pypi.org/search/?q=nvfuser)

Expand Down

0 comments on commit 6f9d811

Please sign in to comment.