Add PyTorch version environment variable, to facilitate SMTT

[yongyanrao]

Issue #, if available:

Description of changes: feature: add PyTorch version environment variable, to facilitate SMTT

Please refer to SMTT PR: https://github.com/aws/sagemaker-training-toolkit/pull/182

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

[satishpasumarthi]

Link SM toolkit PR in the description and viceversa for easier tracking

[sagemaker-bot]

AWS CodeBuild CI Report

• CodeBuild project: sagemaker-pytorch-training-container-pr
• Commit ID: 674608ab1718b39c2c1ed60e3eb3efd9328ed793
• Result: SUCCEEDED
• Build Logs (available for 30 days)

Powered by github-codebuild-logs, available on the AWS Serverless Application Repository

[sagemaker-bot]

AWS CodeBuild CI Report

• CodeBuild project: sagemaker-pytorch-training-toolkit-pr
• Commit ID: 674608ab1718b39c2c1ed60e3eb3efd9328ed793
• Result: SUCCEEDED
• Build Logs (available for 30 days)

Powered by github-codebuild-logs, available on the AWS Serverless Application Repository

[sagemaker-bot]

AWS CodeBuild CI Report

• CodeBuild project: sagemaker-pytorch-training-container-gpu-tests
• Commit ID: 674608ab1718b39c2c1ed60e3eb3efd9328ed793
• Result: SUCCEEDED
• Build Logs (available for 30 days)

Powered by github-codebuild-logs, available on the AWS Serverless Application Repository

[sagemaker-bot]

AWS CodeBuild CI Report

• CodeBuild project: sagemaker-pytorch-training-container-unit-tests
• Commit ID: 674608ab1718b39c2c1ed60e3eb3efd9328ed793
• Result: FAILED
• Build Logs (available for 30 days)

Powered by github-codebuild-logs, available on the AWS Serverless Application Repository

[satishpasumarthi]

Ok to merge if local testing passes.