sagemaker-pytorch-training-toolkit/docker/1.5.0/py3/Dockerfile.gpu at master ยท aws/sagemaker-pytorch-training-toolkit
Toolkit for running PyTorch training scripts on SageMaker. Dockerfiles used for building SageMaker Pytorch Containers are at https://github.com/aws/deep-learning-containers. - aws/sagemaker-pytorch...