Installing SparseZoo

SparseZoo is Neural Magic's repository of pre-sparsified deep learning models across various domains, such as computer vision and NLP. The easiest way to access and use these models is through:

• SparseML: For model optimization, integration into your training pipelines
• DeepSparse: For high-performance deployment of sparsified models

Prerequisites

Hardware Requirements

• CPU: Intel, AMD, or ARM

Software Requirements

• OS: Linux (e.g., Ubuntu, CentOS, Red Hat); MacOS (limited support)
• Python: 3.8 - 3.11

Installation

PyPI

For the most common use case, install the current release version of SparseZoo using PyPi:

pip install sparsezoo-nightly

GitHub (Advanced)

For development purposes or advanced use cases, install directly from the GitHub repository:

pip install git+https://github.com/neuralmagic/sparsezoo.git@main

Or from a locally cloned repository:

git clone https://github.com/neuralmagic/sparsezoo.git
cd sparsezoo
git checkout main
pip install -e .[dev]