Skip to main content

Installation

The Python package is named vajra-streamer, but the import package is vajra.

pip install vajra-streamer

Requirements​

RequirementValue
Python>=3.8
PyTorchtorch>=2.0.0
CUDARequired
Native librarylibvajra.so on Linux or libvajra.dylib on macOS

The package returns CUDA tensors, so a CPU-only PyTorch install is not enough.

Import Check​

from vajra import VajraStreamer, StreamConfig

If this import raises OSError, Python found the vajra package but could not load the native libvajra library. See Errors and Troubleshooting to resolve it.