sagemaker-training==3.9.2
tensorflow==2.6.0
numpy==1.19.5
pillow==8.3.2