Skip to content

Commit

Permalink
update to pytorch and dgl pins used in cugraph repo
Browse files Browse the repository at this point in the history
  • Loading branch information
jameslamb committed Oct 21, 2024
1 parent 887e32f commit 4344fd0
Show file tree
Hide file tree
Showing 8 changed files with 16 additions and 16 deletions.
4 changes: 2 additions & 2 deletions conda/environments/all_cuda-118_arch-x86_64.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ dependencies:
- cupy>=12.0.0
- cython>=3.0.0
- dask-cudf==24.12.*,>=0.0.0a0
- dglteam/label/th23_cu118::dgl
- dglteam/label/th23_cu118::dgl>=2.4.0.th23.cu*
- doxygen
- graphviz
- ipython
Expand All @@ -40,7 +40,7 @@ dependencies:
- pytest-forked
- pytest-xdist
- pytorch-cuda=11.8
- pytorch::pytorch>=2.0,<2.4.0a0
- pytorch::pytorch>=2.3,<2.4.0a0
- pytorch_geometric>=2.5,<2.6
- raft-dask==24.12.*,>=0.0.0a0
- rapids-build-backend>=0.3.0,<0.4.0.dev0
Expand Down
4 changes: 2 additions & 2 deletions conda/environments/all_cuda-121_arch-x86_64.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ dependencies:
- cupy>=12.0.0
- cython>=3.0.0
- dask-cudf==24.12.*,>=0.0.0a0
- dglteam/label/th23_cu121::dgl
- dglteam/label/th23_cu121::dgl>=2.4.0.th23.cu*
- doxygen
- graphviz
- ipython
Expand All @@ -45,7 +45,7 @@ dependencies:
- pytest-forked
- pytest-xdist
- pytorch-cuda=12.1
- pytorch::pytorch>=2.0,<2.4.0a0
- pytorch::pytorch>=2.3,<2.4.0a0
- pytorch_geometric>=2.5,<2.6
- raft-dask==24.12.*,>=0.0.0a0
- rapids-build-backend>=0.3.0,<0.4.0.dev0
Expand Down
4 changes: 2 additions & 2 deletions conda/environments/all_cuda-124_arch-x86_64.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ dependencies:
- cupy>=12.0.0
- cython>=3.0.0
- dask-cudf==24.12.*,>=0.0.0a0
- dglteam/label/th23_cu121::dgl
- dglteam/label/th23_cu121::dgl>=2.4.0.th23.cu*
- doxygen
- graphviz
- ipython
Expand All @@ -45,7 +45,7 @@ dependencies:
- pytest-forked
- pytest-xdist
- pytorch-cuda=12.1
- pytorch::pytorch>=2.0,<2.4.0a0
- pytorch::pytorch>=2.3,<2.4.0a0
- pytorch_geometric>=2.5,<2.6
- raft-dask==24.12.*,>=0.0.0a0
- rapids-build-backend>=0.3.0,<0.4.0.dev0
Expand Down
10 changes: 5 additions & 5 deletions dependencies.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -387,7 +387,7 @@ dependencies:
common:
- output_types: [conda]
packages:
- pytorch::pytorch>=2.0,<2.4.0a0
- pytorch::pytorch>=2.3,<2.4.0a0
- torchdata
- pydantic
specific:
Expand All @@ -404,7 +404,7 @@ dependencies:
matrices:
- matrix: {cuda: "12.*"}
packages:
- &pytorch_pip torch>=2.0,<2.4.0a0
- &pytorch_pip torch>=2.3,<2.4.0a0
- *tensordict
- matrix: {cuda: "11.*"}
packages:
Expand All @@ -426,11 +426,11 @@ dependencies:
matrices:
- matrix: {cuda: "12.*"}
packages:
- "dglteam/label/th23_cu121::dgl"
- "dglteam/label/th23_cu121::dgl>=2.4.0.th23.cu*"
- matrix: {cuda: "11.*"}
packages:
- "dglteam/label/th23_cu118::dgl"
- {matrix: null, packages: ["dglteam/label/th23_cu121::dgl"]}
- "dglteam/label/th23_cu118::dgl>=2.4.0.th23.cu*"
- {matrix: null, packages: ["dglteam/label/th23_cu121::dgl>=2.4.0.th23.cu*"]}

depends_on_pyg:
common:
Expand Down
4 changes: 2 additions & 2 deletions python/cugraph-dgl/conda/cugraph_dgl_dev_cuda-118.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ channels:
dependencies:
- cugraph==24.12.*,>=0.0.0a0
- dgl>=2.4.0.cu*
- dglteam/label/th23_cu118::dgl
- dglteam/label/th23_cu118::dgl>=2.4.0.th23.cu*
- pre-commit
- pydantic
- pylibcugraphops==24.12.*,>=0.0.0a0
Expand All @@ -18,7 +18,7 @@ dependencies:
- pytest-cov
- pytest-xdist
- pytorch-cuda=11.8
- pytorch::pytorch>=2.0,<2.4.0a0
- pytorch::pytorch>=2.3,<2.4.0a0
- tensordict>=0.1.2
- torchdata
name: cugraph_dgl_dev_cuda-118
2 changes: 1 addition & 1 deletion python/cugraph-dgl/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ test = [
"pytest-cov",
"pytest-xdist",
"tensordict>=0.1.2",
"torch>=2.0,<2.4.0a0",
"torch>=2.3,<2.4.0a0",
] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.

[project.urls]
Expand Down
2 changes: 1 addition & 1 deletion python/cugraph-pyg/conda/cugraph_pyg_dev_cuda-118.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ dependencies:
- pytest-cov
- pytest-xdist
- pytorch-cuda=11.8
- pytorch::pytorch>=2.0,<2.4.0a0
- pytorch::pytorch>=2.3,<2.4.0a0
- pytorch_geometric>=2.5,<2.6
- tensordict>=0.1.2
- torchdata
Expand Down
2 changes: 1 addition & 1 deletion python/cugraph-pyg/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ test = [
"pytest-xdist",
"tensordict>=0.1.2",
"torch-geometric>=2.5,<2.6",
"torch>=2.0,<2.4.0a0",
"torch>=2.3,<2.4.0a0",
] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.

[tool.setuptools]
Expand Down

0 comments on commit 4344fd0

Please sign in to comment.