From 98feef65e854cb5dcaf465ae38c7b0fb1b4ebae8 Mon Sep 17 00:00:00 2001 From: James Lamb Date: Wed, 4 Mar 2026 22:07:46 -0600 Subject: [PATCH] include pytorch conda packages in CUDA 13 test env --- conda/environments/all_cuda-131_arch-aarch64.yaml | 1 + conda/environments/all_cuda-131_arch-x86_64.yaml | 1 + dependencies.yaml | 5 ++++- 3 files changed, 6 insertions(+), 1 deletion(-) diff --git a/conda/environments/all_cuda-131_arch-aarch64.yaml b/conda/environments/all_cuda-131_arch-aarch64.yaml index 895754a6fb4..9a4d0d9444f 100644 --- a/conda/environments/all_cuda-131_arch-aarch64.yaml +++ b/conda/environments/all_cuda-131_arch-aarch64.yaml @@ -81,6 +81,7 @@ dependencies: - python-confluent-kafka - python-xxhash - python>=3.11 +- pytorch>=2.10.0 - rapids-build-backend>=0.4.0,<0.5.0 - rapids-dask-dependency==26.4.*,>=0.0.0a0 - rapids-logger==0.2.*,>=0.0.0a0 diff --git a/conda/environments/all_cuda-131_arch-x86_64.yaml b/conda/environments/all_cuda-131_arch-x86_64.yaml index e6ff5f2e770..832fb585f3b 100644 --- a/conda/environments/all_cuda-131_arch-x86_64.yaml +++ b/conda/environments/all_cuda-131_arch-x86_64.yaml @@ -81,6 +81,7 @@ dependencies: - python-confluent-kafka - python-xxhash - python>=3.11 +- pytorch>=2.10.0 - rapids-build-backend>=0.4.0,<0.5.0 - rapids-dask-dependency==26.4.*,>=0.0.0a0 - rapids-logger==0.2.*,>=0.0.0a0 diff --git a/dependencies.yaml b/dependencies.yaml index 2f3e79ff2be..c67901347b3 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -924,7 +924,10 @@ dependencies: cuda: "12.*" packages: - pytorch>=2.4.0 - # TODO: add a 13.x entry here when pytorch has CUDA 13 packages (https://github.com/pytorch/pytorch/issues/159779) + - matrix: + cuda: "13.*" + packages: + - pytorch>=2.10.0 - matrix: packages: test_python_cudf_polars: