[SPARK-46078][PYTHON][TESTS] Upgrade pytorch for Python 3.12

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

This PR proposes to upgrade PyTorch for Python 3.12

This PR is dependent on #45113

### Why are the changes needed?

To use the official releases.

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

No.

### How was this patch tested?

Will be verified in CI. Seems like their support on Mac is not out yet.

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

No.

Closes #45119 from HyukjinKwon/SPARK-46078.

Authored-by: Hyukjin Kwon <gurwls223@apache.org>
Signed-off-by: Hyukjin Kwon <gurwls223@apache.org>

dev/infra/Dockerfile

@@ -133,9 +133,7 @@ RUN apt-get update && apt-get install -y \
     && rm -rf /var/lib/apt/lists/*
 RUN curl -sS https://bootstrap.pypa.io/get-pip.py | python3.12
 # TODO(SPARK-46647) Add unittest-xml-reporting into Python 3.12 image when it supports Python 3.12
-# TODO(SPARK-46078) Use official one instead of nightly build when it's ready
 RUN python3.12 -m pip install $BASIC_PIP_PKGS $CONNECT_PIP_PKGS lxml && \
-    python3.12 -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu && \
-    python3.12 -m pip install torchvision --index-url https://download.pytorch.org/whl/cpu && \
+    python3.12 -m pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu && \
     python3.12 -m pip install torcheval && \
     python3.12 -m pip cache purge