update sagemaker-tensorflow version to 2.1 (#300)

Author: Elizaaaaa

docker/2.1.0/py2/Dockerfile.cpu

@@ -110,7 +110,7 @@ RUN ${PIP} install --no-cache-dir -U \
     mpi4py==3.0.3 \
     opencv-python==4.2.0.32 \
     "cryptography>=2.3" \
-    "sagemaker-tensorflow>=2.0,<2.1" \
+    "sagemaker-tensorflow>=2.1,<2.2" \
     # Let's install TensorFlow separately in the end to avoid
     # the library version to be overwritten
  && ${PIP} install --no-cache-dir -U \

docker/2.1.0/py2/Dockerfile.gpu

@@ -143,7 +143,7 @@ RUN ${PIP} install --no-cache-dir -U \
     mpi4py==3.0.3 \
     opencv-python==4.2.0.32 \
     "cryptography>=2.3" \
-    "sagemaker-tensorflow>=2.0,<2.1" \
+    "sagemaker-tensorflow>=2.1,<2.2" \
     # Let's install TensorFlow separately in the end to avoid
     # the library version to be overwritten
  && ${PIP} install --no-cache-dir -U \

docker/2.1.0/py3/Dockerfile.cpu

@@ -109,7 +109,7 @@ RUN ${PIP} install --no-cache-dir -U \
     awscli \
     mpi4py==3.0.3 \
     opencv-python==4.2.0.32 \
-    "sagemaker-tensorflow>=2.0,<2.1" \
+    "sagemaker-tensorflow>=2.1,<2.2" \
     # Let's install TensorFlow separately in the end to avoid
     # the library version to be overwritten
  && ${PIP} install --no-cache-dir -U \

docker/2.1.0/py3/Dockerfile.gpu

@@ -153,7 +153,7 @@ RUN ${PIP} install --no-cache-dir -U \
     awscli \
     mpi4py==3.0.3 \
     opencv-python==4.2.0.32 \
-    "sagemaker-tensorflow>=2.0,<2.1" \
+    "sagemaker-tensorflow>=2.1,<2.2" \
     # Let's install TensorFlow separately in the end to avoid
     # the library version to be overwritten
  && ${PIP} install --no-cache-dir -U \