From cecafb4734c1ebfae4c2a04d2b966785b9e3bda9 Mon Sep 17 00:00:00 2001
From: Hyunsu Cho <phcho@nvidia.com>
Date: Tue, 24 Sep 2024 13:25:30 -0700
Subject: [PATCH] Update Conda recipes

---
 conda/recipes/cugraph-dgl/meta.yaml | 2 +-
 conda/recipes/cugraph-pyg/meta.yaml | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/conda/recipes/cugraph-dgl/meta.yaml b/conda/recipes/cugraph-dgl/meta.yaml
index ab193897baa..f18feb415d8 100644
--- a/conda/recipes/cugraph-dgl/meta.yaml
+++ b/conda/recipes/cugraph-dgl/meta.yaml
@@ -31,7 +31,7 @@ requirements:
     - pylibcugraphops ={{ minor_version }}
     - tensordict >=0.1.2
     - python
-    - pytorch >=2.0
+    - pytorch >=2.0,<2.4.0a0
     - cupy >=12.0.0
 
 tests:
diff --git a/conda/recipes/cugraph-pyg/meta.yaml b/conda/recipes/cugraph-pyg/meta.yaml
index ba481165d7a..b6e4db5c7dd 100644
--- a/conda/recipes/cugraph-pyg/meta.yaml
+++ b/conda/recipes/cugraph-pyg/meta.yaml
@@ -31,7 +31,7 @@ requirements:
     - numba >=0.57
     - numpy >=1.23,<3.0a0
     - python
-    - pytorch >=2.0
+    - pytorch >=2.0,<2.4.0a0
     - cupy >=12.0.0
     - cugraph ={{ version }}
     - pylibcugraphops ={{ minor_version }}