This guide covers the installation and setup of InfiniMetrics.
- Python: 3.8 or higher
- Compiler: GCC 11.3
- CMake: 3.20+
Install the core dependencies based on your needs:
# For InfiniLM adapter
pip install numpy torch
# For vLLM adapter
pip install vllm
# For data processing
pip install pandasIf you plan to use the hardware testing modules, you need to build the CUDA memory benchmark suite:
cd infinimetrics/hardware/cuda-memory-benchmark
bash build.shNote: This requires:
- CUDA toolkit (compatible with your GPU driver)
- C++ compiler with CUDA support
- CMake 3.20 or higher
After installation, verify your setup:
# Run a simple hardware test
python main.py format_input_comprehensive_hardware.jsonEnsure CUDA_HOME is set:
export CUDA_HOME=/usr/local/cuda
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATHRefer to your accelerator vendor's documentation for driver and toolkit installation.