Use TensorFlow com a Amazon SageMaker - Amazon SageMaker

As traduções são geradas por tradução automática. Em caso de conflito entre o conteúdo da tradução e da versão original em inglês, a versão em inglês prevalecerá.

Use TensorFlow com a Amazon SageMaker

Você pode usar SageMaker a Amazon para treinar e implantar um modelo usando TensorFlow código personalizado. Os TensorFlow estimadores e modelos do SDK do SageMaker Python e os TensorFlow contêineres de SageMaker código aberto facilitam a criação e a execução de um TensorFlow script. SageMaker

Use a TensorFlow versão 1.11 e posterior

Para TensorFlow as versões 1.11 e posteriores, o Amazon SageMaker Python SDK oferece suporte a scripts de treinamento no modo script.

O que você deseja fazer?

Quero treinar um TensorFlow modelo personalizado em SageMaker.

Para ver um exemplo de caderno Jupyter, consulte Treinamento e exibição do modo TensorFlow script.

Para obter a documentação, consulte Treinar um modelo com TensorFlow.

Eu tenho um TensorFlow modelo no SageMaker qual treinei e quero implantá-lo em um endpoint hospedado.

Para obter mais informações, consulte Implantar modelos de TensorFlow serviço.

Tenho um TensorFlow modelo que treinei fora SageMaker e quero implantá-lo em um SageMaker endpoint

Para obter mais informações, consulte Implantação diretamente dos artefatos do modelo.

Quero ver a documentação da API para as classes do Amazon SageMaker Python SDK TensorFlow.

Para obter mais informações, consulte TensorFlow Estimador.

Quero encontrar o repositório do SageMaker TensorFlow contêiner.

Para obter mais informações, consulte GitHub Repositório de SageMaker TensorFlow contêineres.

Quero encontrar informações sobre as TensorFlow versões suportadas pelo AWS Deep Learning Containers.

Para obter mais informações, consulte as Imagens de contêiner de aprendizado profundo disponíveis.

Para obter informações gerais sobre como escrever scripts de treinamento no modo TensorFlow TensorFlow script e usar estimadores e modelos do modo script com SageMaker, consulte Usando TensorFlow com o SDK do Python SageMaker .

Use o Modo TensorFlow Legado para as versões 1.11 e anteriores

O Amazon SageMaker Python SDK fornece um modo legado compatível com TensorFlow as versões 1.11 e anteriores. Use scripts de TensorFlow treinamento do modo legado para executar TensorFlow trabalhos em SageMaker se:

  • Você tiver scripts no modo legado que não deseja converter em modo script.

  • Você deseja usar uma TensorFlow versão anterior à 1.11.

Para obter informações sobre como escrever TensorFlow scripts de modo legado para usar com o SDK do SageMaker Python, consulte TensorFlow SageMaker Estimators and Models.