This repository is tested on Python 3.6+, and Linux/Debian systems.
It is recommended to install in a
to keep your system in order.
Currently supported ML Frameworks are the following:
Install with pip using:
pip install sparseml
Supported Framework Versions
The currently supported framework versions are:
>= 1.1.0, < 1.8.0
2.3.0-tf(through the TensorFlow
2.2package; as of Feb 1st, 2021,
keras2onnxhas not been tested for TensorFlow >=
TensorFlow V1-supported versions: >=
2.Xis not currently supported)
Additionally, optional dependencies can be installed based on the framework you are using.
pip install sparseml[torch]
pip install sparseml[tf_keras]
pip install sparseml[tf_v1]
TensorFlow V1 with GPU operations enabled:
pip install sparseml[tf_v1_gpu]
Depending on your device and CUDA version, you may need to install additional dependencies for using TensorFlow V1 with GPU operations.
You can find these steps here .
Note, TensorFlow V1 is no longer being built for newer operating systems such as Ubuntu 20.04. Therefore, SparseML with TensorFlow V1 is unsupported on these operating systems as well.