fix: pin setuptools to 81.0.0 for pkg_resources compatibility

Author: Bhanu Teja Goshikonda

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

@@ -186,7 +186,7 @@ RUN pip install --no-cache-dir \
     "idna>=3.7" \
     "tqdm>=4.66.3" \
     "requests>=2.32.0" \
-    "setuptools>=80.10.1" \
+    "setuptools==81.0.0" \
     "urllib3>=2.5.0" \
     "awscli" \
     opencv-python==4.11.0.86 \

pytorch/training/docker/2.10/py3/cu130/Dockerfile.gpu

@@ -105,7 +105,7 @@ RUN pip install --no-cache-dir \
     "idna>=3.7" \
     "tqdm>=4.66.3" \
     "requests>=2.32.0" \
-    "setuptools>=80.10.1" \
+    "setuptools==81.0.0" \
     "urllib3>=2.5.0" \
     ninja \
     opencv-python==4.11.0.86 \