Skip to main content

Open source library for using PyTorch to train models on Amazon SageMaker.

Project description

SageMaker PyTorch Training Toolkit is an open-source library for using PyTorch to train models on Amazon SageMaker.

This toolkit depends and extends the base SageMaker Training Toolkit with PyTorch specific support.

For inference, see SageMaker PyTorch Inference Toolkit.

For the Dockerfiles used for building SageMaker PyTorch Containers, see AWS Deep Learning Containers.

For information on running PyTorch jobs on Amazon SageMaker, please refer to the SageMaker Python SDK documentation.

For notebook examples: SageMaker Notebook Examples.

Contributing

Please read CONTRIBUTING.md for details on our code of conduct, and the process for submitting pull requests to us.

License

SageMaker PyTorch Training Toolkit is licensed under the Apache 2.0 License. It is copyright 2018 Amazon .com, Inc. or its affiliates. All Rights Reserved. The license is available at: http://aws.amazon.com/apache2.0/

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

sagemaker_pytorch_training-2.7.0.tar.gz (8.1 kB view details)

Uploaded Source

File details

Details for the file sagemaker_pytorch_training-2.7.0.tar.gz.

File metadata

  • Download URL: sagemaker_pytorch_training-2.7.0.tar.gz
  • Upload date:
  • Size: 8.1 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.3.0 pkginfo/1.8.3 requests/2.28.1 setuptools/47.1.0 requests-toolbelt/0.9.1 tqdm/4.64.0 CPython/3.7.12

File hashes

Hashes for sagemaker_pytorch_training-2.7.0.tar.gz
Algorithm Hash digest
SHA256 3ea89416fa7a14364e7a9de865c103c7ad15f1f0bf3956a8b982cbde39cd6a71
MD5 d923487dae087c658a25b69bdabd7fce
BLAKE2b-256 e7565e3180b7e804177822f747de29ba2b8534162e031bd9765c649c173e0734

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page