Location of Huggingface SageMaker Dockerfile.

Question

Where is the github repository of the Dockerfile for Huggingface training with SageMaker? I see [this](https://github.com/aws/sagemaker-huggingface-inference-toolkit) repository for inference, but do not see one for training.

Answer

There are a bunch of Dockerfiles in the [DLC repo](https://github.com/aws/deep-learning-containers/). Here's the [HuggingFace training Dockerfile for PyTorch 1.9](https://github.com/aws/deep-learning-containers/blob/diy_sm-pt-1.9/huggingface/pytorch/training/docker/1.9/py3/cu111/Dockerfile.gpu).

Answer

HuggingFace Dockerfiles are located

Location of Huggingface SageMaker Dockerfile.

Conteúdo relevante