From 82088e77a3fd896d65f471d87188cb3bfb8b3e93 Mon Sep 17 00:00:00 2001
From: Amir Sadoughi <sadoughi@fb.com>
Date: Mon, 26 Aug 2024 13:39:43 -0400
Subject: [PATCH] bump pytorch to >=2

---
 .github/actions/build_cmake/action.yml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/actions/build_cmake/action.yml b/.github/actions/build_cmake/action.yml
index 8f42af5e0..b9e78ec05 100644
--- a/.github/actions/build_cmake/action.yml
+++ b/.github/actions/build_cmake/action.yml
@@ -63,9 +63,9 @@ 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 pytorch-cuda=12.4 -c pytorch>=2 -c nvidia/label/cuda-12.4.0
         else
-          conda install -y -q pytorch -c pytorch
+          conda install -y -q pytorch -c pytorch>=2
         fi
     - name: ROCm - Install dependencies
       if: inputs.rocm == 'ON'