search

aws / sagemaker-mxnet-training-toolkit

Toolkit for running MXNet training scripts on SageMaker. Dockerfiles used for building SageMaker MXNet Containers are at https://github.com/aws/deep-learning-containers.
Apache License 2.0
60 stars 55 forks source link

Add container support to install_requires #17

Closed ChoiByungWook closed 6 years ago

ChoiByungWook commented 6 years ago

Issue #, if available:

Description of changes: Move container support installation into install_requires.

https://pypi.python.org/pypi/sagemaker-container-support

Image builds locally and passes integ tests.

pytest test/integ/ --docker-base-name 142577830533.dkr.ecr.us-west-2.amazonaws.com/preprod-mxnet --tag 0.12.1-cpu-py2 --framework-version 0.12.1 --py-version 2 --processor cpu

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.