Default Pytorch Python version does not match official build Dockerfile

[plynch-chwy]

Please fill out the form below.

System Information

• Framework: Pytorch
• Framework Version: 1.0.0.dev
• Python Version:
• CPU or GPU: GPU
• Python SDK Version: sagemaker-containers 2.4.4.post2
• Are you using a custom image:

Describe the problem

Official Pytorch image does not match build from sagemaker-pytorch-containerl

Current default sagemaker.pytorch.PyTorch image uses 3.5.2, while "official build" from above repo uses 3.6.

See aws/sagemaker-pytorch-training-toolkit#89

[jesterhazy]

@plynch-chewy

You are getting an out of date image because you are using an unmaintained pre-release version of our pytorch framework. Use framework_version=1.0.0 instead of 1.0.0.dev.