You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+39-8Lines changed: 39 additions & 8 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -13,12 +13,24 @@ NVIDIA Inference Xfer Library (NIXL) is targeted for accelerating point to point
13
13
## Pre-build Distributions
14
14
### PyPI Wheel
15
15
16
-
The nixl python API and libraries, including UCX, are available directly through PyPI:
16
+
The nixl python API and libraries, including UCX, are available directly through PyPI.
17
+
18
+
It can be installed for CUDA 12 with:
19
+
20
+
```
21
+
pip install nixl[cu12]
22
+
```
23
+
24
+
For CUDA 13 with:
17
25
18
26
```
19
-
pip install nixl
27
+
pip install nixl[cu13]
20
28
```
21
29
30
+
For backwards compatibility, `pip install nixl` installs automatically `nixl[cu12]`, continuing to work seamlessly for CUDA 12 users without requiring changes to downstream project dependencies.
31
+
32
+
If both `nixl-cu12` and `nixl-cu13` are installed at the same time in an environment, `nixl-cu13` takes precedence.
33
+
22
34
## Prerequisites for source build
23
35
### Ubuntu:
24
36
@@ -30,7 +42,7 @@ pip install nixl
30
42
31
43
### Python
32
44
33
-
`$ pip3 install meson ninja pybind11`
45
+
`$ pip3 install meson ninja pybind11 tomlkit`
34
46
35
47
### UCX
36
48
@@ -151,18 +163,37 @@ $ ninja
151
163
152
164
NIXL provides Python bindings through pybind11. For detailed Python API documentation, see [docs/python_api.md](docs/python_api.md).
153
165
154
-
The preferred way to install the Python bindings is through pip:
166
+
The preferred way to install the Python bindings is through pip from PyPI:
155
167
156
168
```bash
157
-
pip install nixl
169
+
pip install nixl[cu12]
158
170
```
159
171
160
-
Or build from source:
172
+
Or for CUDA 13 with:
161
173
162
174
```bash
163
-
# From the root nixl directory
175
+
pip install nixl[cu13]
176
+
```
177
+
178
+
To build and install the Python bindings from source, you have to build and install separately the platform-specific package and the `nixl` meta-package:
0 commit comments