Fix GPU image build for mxnet 0.12

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

Fix GPU image build for mxnet 0.12 #10

Closed owen-t closed 6 years ago

owen-t commented 6 years ago

Description of changes: Fix GPU image build for mxnet 0.12

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