Added PyVersions to all SM DDP notebooks (#129)

piyushghai · web-flow · commit 1d00256dfabd · 2020-12-07T12:23:35.000-08:00
* Update py version in TF BERT

* Update TF MaskRCNN

* Update PT BERT

* update PT MaskRCNN
diff --git a/training/distributed_training/pytorch/data_parallel/bert/pytorch_smdataparallel_bert_demo.ipynb b/training/distributed_training/pytorch/data_parallel/bert/pytorch_smdataparallel_bert_demo.ipynb
@@ -240,6 +240,7 @@
     "                    role=role,\n",
     "                    image_uri=docker_image,\n",
     "                    framework_version='1.6.0',\n",
+    "                    py_version='py3',\n",
     "                    instance_count=instance_count,\n",
     "                    instance_type=instance_type,\n",
     "                    sagemaker_session=sagemaker_session,\n",
diff --git a/training/distributed_training/pytorch/data_parallel/maskrcnn/pytorch_smdataparallel_maskrcnn_demo.ipynb b/training/distributed_training/pytorch/data_parallel/maskrcnn/pytorch_smdataparallel_maskrcnn_demo.ipynb
@@ -212,6 +212,8 @@
     "                        source_dir='.',\n",
     "                        instance_count=instance_count,\n",
     "                        instance_type=instance_type,\n",
+    "                        framework_version='1.6.0',\n",
+    "                        py_version='py3',\n",
     "                        sagemaker_session=sagemaker_session,\n",
     "                        hyperparameters=hyperparameters, \n",
     "                        subnets=subnets, \n",
diff --git a/training/distributed_training/tensorflow/data_parallel/bert/tensorflow2_smdataparallel_bert_demo.ipynb b/training/distributed_training/tensorflow/data_parallel/bert/tensorflow2_smdataparallel_bert_demo.ipynb
@@ -245,6 +245,7 @@
     "                        image_uri=docker_image,\n",
     "                        source_dir='deep-learning-models/models/nlp',\n",
     "                        framework_version='2.3.1',\n",
+    "                        py_version='py3',\n",
     "                        instance_count=instance_count,\n",
     "                        instance_type=instance_type,\n",
     "                        sagemaker_session=sagemaker_session,\n",
diff --git a/training/distributed_training/tensorflow/data_parallel/maskrcnn/tensorflow2_smdataparallel_maskrcnn_demo.ipynb b/training/distributed_training/tensorflow/data_parallel/maskrcnn/tensorflow2_smdataparallel_maskrcnn_demo.ipynb
@@ -240,6 +240,7 @@
     "                        image_uri=docker_image,\n",
     "                        source_dir='.',\n",
     "                        framework_version='2.3.1',\n",
+    "                        py_version='py3',\n",
     "                        instance_count=instance_count,\n",
     "                        instance_type=instance_type,\n",
     "                        sagemaker_session=sagemaker_session,\n",
