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: