Update 2.5.0.yaml (pytorch#2498)

Summary: Pull Request resolved: pytorch#2498 Reviewed By: kit1980 Differential Revision: D64407151 Pulled By: atalman fbshipit-source-id: 0637d812144f13dad41b640e70fd65619a183c67
masnesral · Oct 16, 2024 · 9e670cd · 9e670cd
1 parent c396191
commit 9e670cd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/userbenchmark/release-test/configs/2.5.0.yaml b/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

diff --git a/userbenchmark/release-test/setup_env.sh b/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)'