Recomenda-se criar um ambiente virtual Python antes de prosseguir com a instalação. Por favor, veja a instalação TensorFlow guia para 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 detalhes da versão, consulte nossas notas de lançamento .
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 a poda, a página Visão geral é aqui .
Desde TensorFlow não é incluído como uma dependência do pacote TensorFlow modelo de otimização (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 Bazel sistema de compilação.
# 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