Fix torch 2.10 version pinning and remove setuptools pin

Author: Bhanu Teja Goshikonda

pytorch/training/buildspec-2-10-ec2.yml

@@ -44,7 +44,7 @@ images:
   BuildEC2CPUPTTrainPy3DockerImage:
     <<: *TRAINING_REPOSITORY
     build: &PYTORCH_CPU_TRAINING_PY3 false
-    image_size_baseline: 12000
+    image_size_baseline: 7200
     device_type: &DEVICE_TYPE cpu
     python_version: &DOCKER_PYTHON_VERSION py3
     tag_python_version: &TAG_PYTHON_VERSION py313

pytorch/training/buildspec-2-10-sm.yml

@@ -44,7 +44,7 @@ images:
   BuildSageMakerCPUPTTrainPy3DockerImage:
     <<: *TRAINING_REPOSITORY
     build: &PYTORCH_CPU_TRAINING_PY3 false
-    image_size_baseline: 12000
+    image_size_baseline: 7200
     device_type: &DEVICE_TYPE cpu
     python_version: &DOCKER_PYTHON_VERSION py3
     tag_python_version: &TAG_PYTHON_VERSION py313

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==81.0.0" \
+    "setuptools>=80.10.1" \
     "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==81.0.0" \
+    "setuptools>=80.10.1" \
     "urllib3>=2.5.0" \
     ninja \
     opencv-python==4.11.0.86 \