From 2929e81fe03afbb609dd302491ebfce1688a1f99 Mon Sep 17 00:00:00 2001 From: Hyunsu Cho Date: Mon, 23 Sep 2024 11:49:36 -0700 Subject: [PATCH] Add pin pytorch<2.4 --- conda/environments/all_cuda-118_arch-x86_64.yaml | 2 +- conda/environments/all_cuda-125_arch-x86_64.yaml | 2 +- dependencies.yaml | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml index 6473b5c3d9e..000f892a4d2 100644 --- a/conda/environments/all_cuda-118_arch-x86_64.yaml +++ b/conda/environments/all_cuda-118_arch-x86_64.yaml @@ -58,7 +58,7 @@ dependencies: - pytest-mpl - pytest-xdist - python-louvain -- pytorch>=2.0 +- pytorch>=2.0,<2.4 - raft-dask==24.10.*,>=0.0.0a0 - rapids-build-backend>=0.3.1,<0.4.0.dev0 - rapids-dask-dependency==24.10.*,>=0.0.0a0 diff --git a/conda/environments/all_cuda-125_arch-x86_64.yaml b/conda/environments/all_cuda-125_arch-x86_64.yaml index d4f9e7b5fe3..3f65ebae0e4 100644 --- a/conda/environments/all_cuda-125_arch-x86_64.yaml +++ b/conda/environments/all_cuda-125_arch-x86_64.yaml @@ -63,7 +63,7 @@ dependencies: - pytest-mpl - pytest-xdist - python-louvain -- pytorch>=2.0 +- pytorch>=2.0,<2.4 - raft-dask==24.10.*,>=0.0.0a0 - rapids-build-backend>=0.3.1,<0.4.0.dev0 - rapids-dask-dependency==24.10.*,>=0.0.0a0 diff --git a/dependencies.yaml b/dependencies.yaml index 85e6a208957..79f1ffea343 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -714,7 +714,7 @@ dependencies: common: - output_types: [conda] packages: - - &pytorch_unsuffixed pytorch>=2.0 + - &pytorch_unsuffixed pytorch>=2.0,<2.4 - torchdata - pydantic - ogb