Fix package regression in PyTorch 2.7.1 training images

Author: Bhanu Teja Goshikonda

pytorch/training/docker/2.7/py3/Dockerfile.cpu

@@ -178,7 +178,8 @@ RUN pip install --no-cache-dir \
     ipykernel \
     pillow \
     h5py \
-    fsspec \
+    "fsspec>=2025.7.0" \
+    "s3fs>=2025.9.0" \
     "idna>=3.7" \
     "tqdm>=4.66.3" \
     "requests>=2.32.0" \
@@ -188,7 +189,20 @@ RUN pip install --no-cache-dir \
     "opencv-python==4.11.0.86" \
     mpi4py \
     jinja2>=3.1.6 \
-    tornado>=6.5.1
+    tornado>=6.5.1 \
+    "aiobotocore>=2.5.0" \
+    "aiohttp>=3.8.0" \
+    "aioitertools>=0.11.0" \
+    "aiosignal>=1.3.1" \
+    "aiohappyeyeballs>=2.3.0" \
+    "frozenlist>=1.4.0" \
+    "multidict>=6.0.0" \
+    "propcache>=0.1.0" \
+    "yarl>=1.9.1" \
+    "langcodes>=3.3.0" \
+    "language-data>=1.2.0" \
+    "marisa-trie>=1.1.0" \
+    "typer>=0.9.0"
 
 RUN curl -o /license.txt https://aws-dlc-licenses.s3.amazonaws.com/pytorch-2.7/license.txt
 

pytorch/training/docker/2.7/py3/cu128/Dockerfile.gpu

@@ -129,6 +129,7 @@ RUN pip install --no-cache-dir \
     "pillow>=11.3.0" \
     h5py \
     "fsspec>=2025.7.0" \
+    "s3fs>=2025.9.0" \
     "markupsafe>=3.0.2" \
     mpi4py \
     s3torchconnector \
@@ -150,6 +151,19 @@ RUN pip install --no-cache-dir \
     blis \
     "jinja2>=3.1.6"\
     "typing-extensions>=4.14.1" \
+    "aiobotocore>=2.5.0" \
+    "aiohttp>=3.8.0" \
+    "aioitertools>=0.11.0" \
+    "aiosignal>=1.3.1" \
+    "aiohappyeyeballs>=2.3.0" \
+    "frozenlist>=1.4.0" \
+    "multidict>=6.0.0" \
+    "propcache>=0.1.0" \
+    "yarl>=1.9.1" \
+    "langcodes>=3.3.0" \
+    "language-data>=1.2.0" \
+    "marisa-trie>=1.1.0" \
+    "typer>=0.9.0" \
  && pip uninstall -y dataclasses
 
 # Removing the cache as it is needed for security verification
@@ -227,4 +241,4 @@ RUN bash setup_oss_compliance.sh ${PYTHON} && rm setup_oss_compliance.sh
 RUN rm -rf /root/.cache | true
 
 ENTRYPOINT ["bash", "-m", "start_with_right_hostname.sh"]
-CMD ["/bin/bash"]
+CMD ["/bin/bash"]