Skip to content

Commit 09d1a83

Browse files
committed
chore(deps): request specific gpu torch build
1 parent 06ed784 commit 09d1a83

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

dependencies.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -374,7 +374,7 @@ dependencies:
374374
- matrix:
375375
cuda: "13.*"
376376
packages:
377-
- &pytorch_pip torch>=2.9.0
377+
- &pytorch_pip torch==2.10.0+cu130
378378
- matrix:
379379
packages:
380380
- *pytorch_pip

python/cugraph-pyg/pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ test = [
5656
"pytest-cov",
5757
"pytest-xdist",
5858
"sentence-transformers",
59-
"torch>=2.9.0",
59+
"torch==2.10.0+cu130",
6060
] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
6161

6262
[tool.setuptools.dynamic]

0 commit comments

Comments
 (0)