Skip to content

Commit

Permalink
Updating readme (#3365)
Browse files Browse the repository at this point in the history
Fixes: #1078 

updating wheel publish and supported python versions
  • Loading branch information
jjsjann123 authored Nov 21, 2024
1 parent 3266b9d commit 9effdc6
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -11,12 +11,12 @@ A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
## Installation

We publish nightly wheel packages on https://pypi.nvidia.com, while build against stable torch version on https://pypi.org.
**Wheels are published for Python version: _3.10_, _3.12_**.

built-env | cuda 12.1
:---: | :---:
torch 2.3 | nvfuser-cu121-torch23
torch 2.4 | nvfuser-cu121-torch24
torch nightly wheel | nvfuser-cu121
built-env | cuda 11.8 | cuda 12.1 | cuda 12.4
:---: | :---: | :---: | :---:
torch 2.5 (pypi.org)| nvfuser-cu118-torch25 | nvfuser-cu121-torch25 | nvfuser-cu124-torch25
torch nightly (pypi.nvidia.com) | nvfuser-cu118 | nvfuser-cu121 | nvfuser-cu124

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.

Expand Down

0 comments on commit 9effdc6

Please sign in to comment.