Skip to content

Commit f96938c

Browse files
committed
chore(deps): pin explicit cuda version of torch in tests
1 parent 09d1a83 commit f96938c

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

ci/test_wheel_pylibwholegraph.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ rapids-pip-retry install \
3434
--extra-index-url ${PYTORCH_INDEX} \
3535
"$(echo "${PYLIBWHOLEGRAPH_WHEELHOUSE}"/pylibwholegraph*.whl)[test]" \
3636
"${LIBWHOLEGRAPH_WHEELHOUSE}"/*.whl \
37-
'torch>=2.3'
37+
'torch==2.10.0+cu130'
3838

3939
rapids-logger "pytest pylibwholegraph"
4040
cd python/pylibwholegraph/pylibwholegraph/tests

dependencies.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -370,7 +370,7 @@ dependencies:
370370
- matrix:
371371
cuda: "12.*"
372372
packages:
373-
- torch>=2.3
373+
- torch==2.10.0+cu130
374374
- matrix:
375375
cuda: "13.*"
376376
packages:

0 commit comments

Comments
 (0)