feat: add 2.0 for torch_distributed in training

Shibo Xing · Shibo Xing · commit d93311d2e987 · 2023-04-18T17:53:16.000Z
diff --git a/src/sagemaker/fw_utils.py b/src/sagemaker/fw_utils.py
@@ -153,7 +153,7 @@
 ]
 
 
-TORCH_DISTRIBUTED_GPU_SUPPORTED_FRAMEWORK_VERSIONS = ["1.13.1"]
+TORCH_DISTRIBUTED_GPU_SUPPORTED_FRAMEWORK_VERSIONS = ["1.13.1", "2.0.0"]
 
 TRAINIUM_SUPPORTED_DISTRIBUTION_STRATEGIES = ["torch_distributed"]
 TRAINIUM_SUPPORTED_TORCH_DISTRIBUTED_FRAMEWORK_VERSIONS = [

Original file line number	Diff line number	Diff line change
`@@ -153,7 +153,7 @@`
`153`	`153`	`]`
`154`	`154`
`155`	`155`
`156`		`-TORCH_DISTRIBUTED_GPU_SUPPORTED_FRAMEWORK_VERSIONS = ["1.13.1"]`
	`156`	`+TORCH_DISTRIBUTED_GPU_SUPPORTED_FRAMEWORK_VERSIONS = ["1.13.1", "2.0.0"]`
`157`	`157`
`158`	`158`	`TRAINIUM_SUPPORTED_DISTRIBUTION_STRATEGIES = ["torch_distributed"]`
`159`	`159`	`TRAINIUM_SUPPORTED_TORCH_DISTRIBUTED_FRAMEWORK_VERSIONS = [`