To install the latest stable release of MCT from PyPi, follow the MCT install guide and run the following command:
pip install model-compression-toolkit
To work with the MCT source code, follow these steps:
- Clone the repository:
git clone https://github.com/SonySemiconductorSolutions/mct-model-optimization.git
cd mct-model-optimization
- Install the required packages listed in the requirements file:
pip install -r requirements.txt
Before running MCT, make sure to install one of the supported frameworks: TensorFlow or PyTorch.
If you intend to use MCT with TensorFlow, install the following package: tensorflow
If you plan to use MCT with PyTorch, install the following package: torch
If you encounter any issues during installation, please open an issue.