Recomenda-se criar um ambiente virtual Python antes de prosseguir com a instalação. Consulte o guia de instalação do TensorFlow para obter mais informações.
Construções estáveis
Para instalar a versão mais recente, execute o seguinte:
# Installing with the `--upgrade` flag ensures you'll get the latest version.
pip install --user --upgrade tensorflow-model-optimization
Para obter detalhes da versão, consulte nossas notas de versão .
Para obter a versão necessária do TensorFlow e outras informações de compatibilidade, consulte a seção Matriz de compatibilidade da API da página Visão geral da técnica que você pretende usar. Por exemplo, para poda, a página Visão geral está aqui .
Como o TensorFlow não está incluído como uma dependência do pacote TensorFlow Model Optimization (em setup.py ), você deve instalar explicitamente o pacote TensorFlow ( tf-nightly ou tf-nightly-gpu ). Isso nos permite manter um pacote em vez de pacotes separados para TensorFlow habilitado para CPU e GPU.
Instalando da fonte
Você também pode instalar a partir da fonte. Isso requer o sistema de compilação Bazel .
# To install dependencies on Ubuntu:
# sudo apt-get install bazel git python-pip
# For other platforms, see Bazel docs above.
git clone https://github.com/tensorflow/model-optimization.git
cd model-optimization
bazel build --copt=-O3 --copt=-march=native :pip_pkg
PKGDIR=$(mktemp -d)
./bazel-bin/pip_pkg $PKGDIR
pip install --user --upgrade $PKGDIR/*.whl