From 03e96034212aef603060966f490a039385aa5d72 Mon Sep 17 00:00:00 2001 From: Yury Date: Wed, 24 Jan 2024 23:32:49 +0200 Subject: [PATCH] Use faiss-gpu on linux with nvidia --- server/requirements-cuda.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/server/requirements-cuda.txt b/server/requirements-cuda.txt index 1bca54c17..7c553b36c 100644 --- a/server/requirements-cuda.txt +++ b/server/requirements-cuda.txt @@ -7,7 +7,8 @@ --extra-index-url https://download.pytorch.org/whl/cu118 torch torchaudio -faiss-cpu +faiss-cpu; sys_platform!='linux' +faiss-gpu; sys_platform=='linux' tensorboard tensorboardx