Update 2.5.0.yaml (#2498)

Summary: Pull Request resolved: #2498

Reviewed By: kit1980

Differential Revision: D64407151

Pulled By: atalman

fbshipit-source-id: 0637d812144f13dad41b640e70fd65619a183c67

Author: juliagmt-google

userbenchmark/release-test/configs/2.5.0.yaml

@@ -1,6 +1,6 @@
 cuda:
-  - version: 12.1
-    magma_version: magma-cuda121
+  - version: 12.4
+    magma_version: magma-cuda124
 pytorch:
   - version: 2.4.1
     conda_channel: pytorch

userbenchmark/release-test/setup_env.sh

@@ -40,7 +40,7 @@ conda install -y -c pytorch ${MAGMA_VERSION}
 if [[ ${PYTORCH_CHANNEL} == "pytorch-test" ]]; then
     pip3 install torch==${PYTORCH_VERSION} torchvision --index-url https://download.pytorch.org/whl/test/cu${CUDA_VERSION//./}
 else
-    pip3 install torch torchvision
+    pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//./}
 fi
 
 python -c 'import torch; print(torch.__version__); print(torch.version.git_version)'