Sionna Installation#
Sionna is only required for development purposes. It is not required to run the OpenAirInterface stack.
Note
Sionna 2.0 requires PyTorch. See the platform-specific sections below.
Virtual Environment#
We recommend using a virtual environment to isolate Python dependencies:
python3 -m venv ~/.venv/sionna-rk
source ~/.venv/sionna-rk/bin/activate
To activate the environment automatically, add to your ~/.profile:
echo 'source ~/.venv/sionna-rk/bin/activate' >> ~/.profile
DGX Spark#
On DGX Spark, install all requirements (including PyTorch and Sionna):
pip install -r requirements.txt
Jetson Thor#
On Jetson Thor, install all requirements:
pip install -r requirements_thor.txt
Jetson AGX Orin & Orin Nano#
On Jetson Orin platforms, install all requirements:
pip install -r requirements_orin.txt
Note
For Jetson platforms, NVIDIA provides pre-built PyTorch wheels via the Jetson PyTorch containers. Refer to the NVIDIA documentation for the latest compatible PyTorch version for your JetPack release.
TensorRT Python Bindings#
To access system TensorRT bindings in the virtual environment:
On AGX Orin:
echo 'export PYTHONPATH=$PYTHONPATH:/usr/lib/python3.10/dist-packages' >> ~/.profile
source ~/.profile
On AGX Thor:
echo 'export PYTHONPATH=$PYTHONPATH:/usr/lib/python3.12/dist-packages' >> ~/.profile
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.profile
source ~/.profile
Verification#
Verify that PyTorch detects the GPU:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.cuda.get_device_name(0))
Expected output (version string, CUDA availability, device name):
2.x.x
True
NVIDIA <device name>