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.