From 02afb377a14c0640e63f26f96bc6bd71c05342fd Mon Sep 17 00:00:00 2001
From: Amir Sadoughi <sadoughi@fb.com>
Date: Tue, 22 Oct 2024 12:50:44 -0400
Subject: [PATCH] test RAFT CI with pytorch 2.4.1

---
 .github/actions/build_cmake/action.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/actions/build_cmake/action.yml b/.github/actions/build_cmake/action.yml
index 2e32dc1578..1ec1186484 100644
--- a/.github/actions/build_cmake/action.yml
+++ b/.github/actions/build_cmake/action.yml
@@ -63,7 +63,7 @@ runs:
           : # skip torch install via conda, we need to install via pip to get
             #  ROCm-enabled version until it's supported in conda by PyTorch
         elif [ "${{ inputs.gpu }}" = "ON" ]; then
-          conda install -y -q pytorch pytorch-cuda=12.4 -c pytorch -c nvidia/label/cuda-12.4.0
+          conda install -y -q "pytorch<2.5" pytorch-cuda=12.4 -c pytorch -c nvidia/label/cuda-12.4.0
         else
           conda install -y -q pytorch -c pytorch
         fi