[SPARK-49862][INFRA][FOLLOWUP] Update dev/infra/Dockerfile too

### What changes were proposed in this pull request?

This is a follow-up of the following to update `dev/infra/Dockerfile` together.
- #50539

### Why are the changes needed?

The original goal of SPARK-49862 was updating both files.

### Does this PR introduce _any_ user-facing change?

No behavior change because this is an infra change.

### How was this patch tested?

Pass the CIs.

### Was this patch authored or co-authored using generative AI tooling?

No.

Closes #52952 from dongjoon-hyun/SPARK-49862.

Authored-by: Dongjoon Hyun <[email protected]>
Signed-off-by: Dongjoon Hyun <[email protected]>
(cherry picked from commit 5c11509)
Signed-off-by: Dongjoon Hyun <[email protected]>

Author: dongjoon-hyun

dev/infra/Dockerfile

@@ -147,9 +147,10 @@ RUN apt-get update && apt-get install -y \
     python3.13 \
     && rm -rf /var/lib/apt/lists/*
 RUN curl -sS https://bootstrap.pypa.io/get-pip.py | python3.13
-# TODO(SPARK-49862) Add BASIC_PIP_PKGS and CONNECT_PIP_PKGS to Python 3.13 image when it supports Python 3.13
 RUN python3.13 -m pip install --ignore-installed 'blinker>=1.6.2' # mlflow needs this
-RUN python3.13 -m pip install 'numpy>=2.1' 'pyarrow>=18.0.0' 'six==1.16.0' 'pandas==2.3.3' scipy coverage matplotlib openpyxl 'grpcio==1.76.0' 'grpcio-status==1.76.0' lxml jinja2 && \
+RUN python3.13 -m pip install $BASIC_PIP_PKGS unittest-xml-reporting $CONNECT_PIP_PKGS lxml && \
+    python3.13 -m pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu && \
+    python3.13 -m pip install torcheval && \
     python3.13 -m pip cache purge
 
 # Remove unused installation packages to free up disk space