add pt 171 to smddp supported frameworks

ChaiBapchya · ChaiBapchya · commit aa90136acece · 2020-12-14T13:44:04.000-08:00
diff --git a/src/sagemaker/fw_utils.py b/src/sagemaker/fw_utils.py
@@ -58,7 +58,7 @@
 )
 SM_DATAPARALLEL_SUPPORTED_FRAMEWORK_VERSIONS = {
     "tensorflow": ["2.3.0", "2.3.1"],
-    "pytorch": ["1.6.0"],
+    "pytorch": ["1.6.0", "1.7.1"],
 }
 SMDISTRIBUTED_SUPPORTED_STRATEGIES = ["dataparallel", "modelparallel"]
 

Original file line number	Diff line number	Diff line change
`@@ -58,7 +58,7 @@`
`58`	`58`	`)`
`59`	`59`	`SM_DATAPARALLEL_SUPPORTED_FRAMEWORK_VERSIONS = {`
`60`	`60`	`"tensorflow": ["2.3.0", "2.3.1"],`
`61`		`- "pytorch": ["1.6.0"],`
	`61`	`+ "pytorch": ["1.6.0", "1.7.1"],`
`62`	`62`	`}`
`63`	`63`	`SMDISTRIBUTED_SUPPORTED_STRATEGIES = ["dataparallel", "modelparallel"]`
`64`	`64`