From 9a75323dec363f2acd3cef79e19ab0bc61b3ad73 Mon Sep 17 00:00:00 2001 From: Vyas Ramasubramani Date: Wed, 7 Dec 2022 16:39:47 -0800 Subject: [PATCH] Ensure that we get the cuda version of faiss. --- conda/environments/all_cuda-115_arch-x86_64.yaml | 2 +- dependencies.yaml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/conda/environments/all_cuda-115_arch-x86_64.yaml b/conda/environments/all_cuda-115_arch-x86_64.yaml index a36747d38d..c87ed198f6 100644 --- a/conda/environments/all_cuda-115_arch-x86_64.yaml +++ b/conda/environments/all_cuda-115_arch-x86_64.yaml @@ -22,7 +22,7 @@ dependencies: - doxygen>=1.8.20 - faiss-proc=*=cuda - gcc_linux-64=9.* -- libfaiss>=1.7.0 +- libfaiss>=1.7.0=cuda* - ninja - pytest - rapids-build-env=23.02.* diff --git a/dependencies.yaml b/dependencies.yaml index 5906a8a428..f55310ffa8 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -114,7 +114,7 @@ dependencies: - ucx>=1.13.0 - ucx-py=0.30.* - ucx-proc=*=gpu - - libfaiss>=1.7.0 + - libfaiss>=1.7.0=cuda* - faiss-proc=*=cuda - dask-cuda=23.02.* test_python: