How to check smdistributed-modelparallel version?

0

According to the doc ( https://sagemaker.readthedocs.io/en/stable/api/training/smd_model_parallel_general.html ), there are different parameters depending on the version of smdistributed-modelparallel module / package. However, I am unable to find a way to check the version (e.g. via sagemaker python SDK) or just from the training container documentation (e.g. https://github.com/aws/deep-learning-containers/blob/master/available_images.md#huggingface-training-containers ).

Any idea?

Thanks!

已提问 2 年前426 查看次数
1 回答
0
已接受的回答

Have not yet found a programmatic way to check the version.

However, for each DLC (Deep Learning Container) available at https://github.com/aws/deep-learning-containers/blob/master/available_images.md , we can look at the corresponding docker build files.

E.g. for PyTorch 1.10.2 with HuggingFace transformers DLC, the corresponding dockerfile is here: https://github.com/aws/deep-learning-containers/blob/master/huggingface/pytorch/training/docker/1.10/py3/cu113/Dockerfile.gpu

And we can see that the version: smdistributed_modelparallel-1.8.1-cp38-cp38-linux_x86_64.whl.

已回答 2 年前

您未登录。 登录 发布回答。

一个好的回答可以清楚地解答问题和提供建设性反馈,并能促进提问者的职业发展。

回答问题的准则