NVSHMEM4Py
NVSHMEM4Py exposes NVSHMEM host, device, and on-stream interfaces to Python, with interoperability for CUDA Python, Numba, PyTorch, CuPy, and CuTe DSL.
Quick Start
Follow the NVSHMEM4Py installation guide to install the package matching the CUDA
major version. This example requires two supported GPUs on one Linux node and mpi4py built for the
active MPI installation.
From the repository root:
mpirun -np 2 python nvshmem4py/examples/hello.py
The output includes one line from each PE, in either order:
Hello from PE 0 of 2
Hello from PE 1 of 2
The example selects one GPU per local MPI rank and initializes NVSHMEM from
MPI_COMM_WORLD. See the main NVSHMEM quick start for platform
requirements and the launching guide for other launchers.
Building Wheels
Wheel targets are part of the NVSHMEM CMake build. See the NVSHMEM4Py CMake configuration for supported options and defaults. For example, to build a Python 3.12 wheel for CUDA 12:
cmake -S . -B build \
-DNVSHMEM4PY_BUILD_ALL_WHEELS=OFF \
-DNVSHMEM4PY_PYTHON_VERSIONS=3.12 \
-DNVSHMEM4PY_CUDA_VERSIONS=12
cmake --build build --target build_nvshmem4py_wheel_cu12_3.12
The wheel is written to build/dist/ and can be installed as described in the
NVSHMEM4Py installation guide.
Examples
The examples cover host, device, on-stream, and framework-interoperability workflows. See the main NVSHMEM README for additional documentation and project information.
Metadata
Release files for nvshmem4py-cu12 0.4.1
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Built distributions (wheels)
Total release size: 58.3 MB
Release files / nvshmem4py_cu12-0.4.1-cp314-cp314-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_34_x86_64.whl
| Download URL | nvshmem4py_cu12-0.4.1-cp314-cp314-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_34_x86_64.whl |
|---|---|
| Size | 4.9 MB |
| Tags | CPython 3.14 Linux glibc 2.17+ x86-64 Linux glibc 2.34+ x86-64 |
|
SHA-256 checksum How to use checksums |
eff73a3964b8f6dcfa54ad781d17638224c9902de0bf551841b25dcfc9a05d2a
|
|
BLAKE2b-256 checksum How to use checksums |
2b0cda033648728f2b76ee67224328d0cd50a57e341e7e7c23f1d2bf207d03af
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.8
|
Release files / nvshmem4py_cu12-0.4.1-cp314-cp314-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_34_aarch64.whl
| Download URL | nvshmem4py_cu12-0.4.1-cp314-cp314-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_34_aarch64.whl |
|---|---|
| Size | 5.0 MB |
| Tags | CPython 3.14 Linux glibc 2.17+ ARM64 Linux glibc 2.34+ ARM64 |
|
SHA-256 checksum How to use checksums |
fda0f7670a2565198d98f90a10985d84198132a4cbd19bd9ff818a0538d15ae4
|
|
BLAKE2b-256 checksum How to use checksums |
28cc7d94108c19b11c539cc1a534f9299205a0fc7e052a3ade19497d265d3fcf
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.8
|
Release files / nvshmem4py_cu12-0.4.1-cp313-cp313-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_34_x86_64.whl
| Download URL | nvshmem4py_cu12-0.4.1-cp313-cp313-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_34_x86_64.whl |
|---|---|
| Size | 5.0 MB |
| Tags | CPython 3.13 Linux glibc 2.17+ x86-64 Linux glibc 2.34+ x86-64 |
|
SHA-256 checksum How to use checksums |
c048c7c42a2b059f03dcaa7a25621c7ad0423b26dac9812c9252648681edff37
|
|
BLAKE2b-256 checksum How to use checksums |
2647c0ddce46fa1443ac53d084b78a2b792e983b7516be67f34c19797b5aa0e9
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.8
|
Release files / nvshmem4py_cu12-0.4.1-cp313-cp313-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_34_aarch64.whl
| Download URL | nvshmem4py_cu12-0.4.1-cp313-cp313-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_34_aarch64.whl |
|---|---|
| Size | 5.0 MB |
| Tags | CPython 3.13 Linux glibc 2.17+ ARM64 Linux glibc 2.34+ ARM64 |
|
SHA-256 checksum How to use checksums |
5b5a498fe9ee63762cc1475aaea5d1f847f5c7107d6430543b8fdcb95d5502cc
|
|
BLAKE2b-256 checksum How to use checksums |
c89385ae8348c3e021a326a24f92318f0c11b04e9d9785585307c6c3d7cb2aee
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.8
|
Release files / nvshmem4py_cu12-0.4.1-cp312-cp312-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_34_x86_64.whl
| Download URL | nvshmem4py_cu12-0.4.1-cp312-cp312-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_34_x86_64.whl |
|---|---|
| Size | 5.0 MB |
| Tags | CPython 3.12 Linux glibc 2.17+ x86-64 Linux glibc 2.34+ x86-64 |
|
SHA-256 checksum How to use checksums |
2be8f8a7dd2a7b1222ce37734a844e08ae0f73fcd2792da0dd1ac5daaf004bd1
|
|
BLAKE2b-256 checksum How to use checksums |
d4ff9032117096b262481be2f43af6fdd4483739ee3e3f861a7e636264fbfa88
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.8
|
Release files / nvshmem4py_cu12-0.4.1-cp312-cp312-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_34_aarch64.whl
| Download URL | nvshmem4py_cu12-0.4.1-cp312-cp312-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_34_aarch64.whl |
|---|---|
| Size | 4.8 MB |
| Tags | CPython 3.12 Linux glibc 2.17+ ARM64 Linux glibc 2.34+ ARM64 |
|
SHA-256 checksum How to use checksums |
3e44cd2cbbbee6692ae24f7fc0378dc4bc56e96a16aef50d2c40f35a1de7f352
|
|
BLAKE2b-256 checksum How to use checksums |
576dcf0ba9f858ebfa1f5ba12f5f3e2c92b0643f9d307be71d6ba7145be4a687
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.8
|
Release files / nvshmem4py_cu12-0.4.1-cp311-cp311-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_34_x86_64.whl
| Download URL | nvshmem4py_cu12-0.4.1-cp311-cp311-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_34_x86_64.whl |
|---|---|
| Size | 4.7 MB |
| Tags | CPython 3.11 Linux glibc 2.17+ x86-64 Linux glibc 2.34+ x86-64 |
|
SHA-256 checksum How to use checksums |
a77091e773ecaa9f57d2eb4035aab9d654b91e364ca1fd4c7dd8644818605cb0
|
|
BLAKE2b-256 checksum How to use checksums |
77e8fce8094fe85cf1e7a12a8cfa2e6b12eb357c90c708840dd5aa9ae3736052
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.8
|
Release files / nvshmem4py_cu12-0.4.1-cp311-cp311-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_34_aarch64.whl
| Download URL | nvshmem4py_cu12-0.4.1-cp311-cp311-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_34_aarch64.whl |
|---|---|
| Size | 4.7 MB |
| Tags | CPython 3.11 Linux glibc 2.17+ ARM64 Linux glibc 2.34+ ARM64 |
|
SHA-256 checksum How to use checksums |
2db694a1a8dec53a432678b1737069e8a8cfacf2c92b92d78862f3ffdd131880
|
|
BLAKE2b-256 checksum How to use checksums |
e4894ab1050ed511bfc1146e70eac7d2cfbb360473b45c8f4e73a6d40c39bc04
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.8
|
Release files / nvshmem4py_cu12-0.4.1-cp310-cp310-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_34_x86_64.whl
| Download URL | nvshmem4py_cu12-0.4.1-cp310-cp310-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_34_x86_64.whl |
|---|---|
| Size | 4.8 MB |
| Tags | CPython 3.10 Linux glibc 2.17+ x86-64 Linux glibc 2.34+ x86-64 |
|
SHA-256 checksum How to use checksums |
dceaff1fb43d5a09d8e08b1955c9d312450d8b62ada8194a09a760ffc93b6d24
|
|
BLAKE2b-256 checksum How to use checksums |
8c22d044ab2989390a16f110a5151c3bf7605eb28809e07cc93bac510499dae4
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.8
|
Release files / nvshmem4py_cu12-0.4.1-cp310-cp310-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_34_aarch64.whl
| Download URL | nvshmem4py_cu12-0.4.1-cp310-cp310-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_34_aarch64.whl |
|---|---|
| Size | 4.9 MB |
| Tags | CPython 3.10 Linux glibc 2.17+ ARM64 Linux glibc 2.34+ ARM64 |
|
SHA-256 checksum How to use checksums |
20d86e7c665de410999d89eddb6a48ec2c5452e5f1be57554ab37245634e7ae6
|
|
BLAKE2b-256 checksum How to use checksums |
bc227b5622132c29b7e4d9db33de5100088b940c5436e8696d328f02dba4b857
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.8
|
Release files / nvshmem4py_cu12-0.4.1-cp39-cp39-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_34_x86_64.whl
| Download URL | nvshmem4py_cu12-0.4.1-cp39-cp39-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_34_x86_64.whl |
|---|---|
| Size | 4.8 MB |
| Tags | CPython 3.9 Linux glibc 2.17+ x86-64 Linux glibc 2.34+ x86-64 |
|
SHA-256 checksum How to use checksums |
bda4def2bbf81d013adfe7df4c39282395b834293df2847078e946e47ee3d7d8
|
|
BLAKE2b-256 checksum How to use checksums |
e1d0e3be334a3b84de7572b8494e8fd7c5e04ac10e79c641a108c37d7fe341bf
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.8
|
Release files / nvshmem4py_cu12-0.4.1-cp39-cp39-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_34_aarch64.whl
| Download URL | nvshmem4py_cu12-0.4.1-cp39-cp39-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_34_aarch64.whl |
|---|---|
| Size | 4.8 MB |
| Tags | CPython 3.9 Linux glibc 2.17+ ARM64 Linux glibc 2.34+ ARM64 |
|
SHA-256 checksum How to use checksums |
374f227353e8c3dec68dd5c49b7c584cc4f398451055c099c2e5b63e1b34617d
|
|
BLAKE2b-256 checksum How to use checksums |
6cf83c7474be50d2307d0c4c9e68878bb4b2d7c3df9c6c5844e3a91214ed6596
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.8
|
