use just pytorch instead of pytorch-gpu

rapidsai · Jan 6, 2025 · 4992680 · 4992680
1 parent 8e5a0f8
commit 4992680
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 18 deletions.
diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml
@@ -38,7 +38,7 @@ dependencies:
 - pytest-cov
 - pytest-forked
 - pytest-xdist
-- pytorch-gpu>=2.3=*cuda118*
+- pytorch>=2.3
 - pytorch_geometric>=2.5,<2.6
 - rapids-build-backend>=0.3.0,<0.4.0.dev0
 - rmm==25.2.*,>=0.0.0a0

diff --git a/conda/environments/all_cuda-121_arch-x86_64.yaml b/conda/environments/all_cuda-121_arch-x86_64.yaml
@@ -44,7 +44,7 @@ dependencies:
 - pytest-cov
 - pytest-forked
 - pytest-xdist
-- pytorch-gpu>=2.3=*cuda120*
+- pytorch>=2.3
 - pytorch_geometric>=2.5,<2.6
 - rapids-build-backend>=0.3.0,<0.4.0.dev0
 - rmm==25.2.*,>=0.0.0a0

diff --git a/conda/environments/all_cuda-124_arch-x86_64.yaml b/conda/environments/all_cuda-124_arch-x86_64.yaml
@@ -44,7 +44,7 @@ dependencies:
 - pytest-cov
 - pytest-forked
 - pytest-xdist
-- pytorch-gpu>=2.3=*cuda120*
+- pytorch>=2.3
 - pytorch_geometric>=2.5,<2.6
 - rapids-build-backend>=0.3.0,<0.4.0.dev0
 - rmm==25.2.*,>=0.0.0a0

diff --git a/dependencies.yaml b/dependencies.yaml
@@ -398,6 +398,7 @@ dependencies:
     common:
       - output_types: [conda]
         packages:
+          - pytorch>=2.3
           - torchdata
           - pydantic
     specific:
@@ -421,19 +422,6 @@ dependencies:
               - *pytorch_pip
               - *tensordict
           - {matrix: null, packages: [*pytorch_pip, *tensordict]}
-      - output_types: [conda]
-        # PyTorch will stop publishing conda packages after 2.5.
-        # Consider switching to conda-forge::pytorch-gpu.
-        # Note that the CUDA version may differ from the official PyTorch wheels.
-        matrices:
-          - matrix: {cuda: "12.*"}
-            packages:
-            - pytorch-gpu>=2.3=*cuda120*
-          - matrix: {cuda: "11.*"}
-            packages:
-            - pytorch-gpu>=2.3=*cuda118*
-          - matrix:
-            packages:
 
   depends_on_dgl:
     specific:

diff --git a/python/cugraph-dgl/conda/cugraph_dgl_dev_cuda-118.yaml b/python/cugraph-dgl/conda/cugraph_dgl_dev_cuda-118.yaml
@@ -16,7 +16,7 @@ dependencies:
 - pytest-benchmark
 - pytest-cov
 - pytest-xdist
-- pytorch-gpu>=2.3=*cuda118*
+- pytorch>=2.3
 - tensordict>=0.1.2
 - torchdata
 name: cugraph_dgl_dev_cuda-118
diff --git a/python/cugraph-pyg/conda/cugraph_pyg_dev_cuda-118.yaml b/python/cugraph-pyg/conda/cugraph_pyg_dev_cuda-118.yaml
@@ -14,7 +14,7 @@ dependencies:
 - pytest-benchmark
 - pytest-cov
 - pytest-xdist
-- pytorch-gpu>=2.3=*cuda118*
+- pytorch>=2.3
 - pytorch_geometric>=2.5,<2.6
 - tensordict>=0.1.2
 - torchdata