| # Base dependencies for export, distillation, evaluation, and plotting. | |
| # Install with: pip install -r requirements.txt | |
| # On Jetson, install NVIDIA's PyTorch first and skip the torch line if it is already present. | |
| numpy | |
| tqdm | |
| PyYAML | |
| matplotlib | |
| opencv-python | |
| lightning | |
| onnx | |
| torch==2.8.0 | |