Esta página foi traduzida pela API Cloud Translation.
Switch to English

Instale a otimização do modelo TensorFlow

É recomendável criar um ambiente virtual Python antes de prosseguir com a instalação. Consulte o guia de instalação do TensorFlow 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 obter detalhes sobre o lançamento, consulte nossas notas de lançamento .

Para a versão necessária do TensorFlow e outras informações de compatibilidade, consulte a seção Matriz de compatibilidade de API da página Visão geral para a 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 o TensorFlow habilitado para CPU e GPU.

Instalando da fonte

Você também pode instalar da fonte. Isso requer o sistema de compilação do 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