Use PyTorch 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 PyTorch com a Amazon SageMaker

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

O que você deseja fazer?

Quero treinar um PyTorch modelo personalizado em SageMaker.

Para obter um exemplo de caderno Jupyter, consulte o caderno de PyTorch exemplo no repositório Amazon SageMaker Examples GitHub.

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

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

Para obter mais informações, consulte Implantar PyTorch modelos.

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

Para obter mais informações, consulte Implantar seu próprio PyTorch modelo.

Quero ver a API documentação das classes do Amazon SageMaker Python SDK PyTorch.

Para obter mais informações, consulte PyTorch Classes.

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

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

Quero encontrar informações sobre as PyTorch 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 PyTorch treinamento e usar PyTorch estimadores e modelos com SageMaker, consulte Usando PyTorch com o Python SageMaker . SDK