From 54d331ee2227a54295e72dd8f98ffac9e12d8025 Mon Sep 17 00:00:00 2001 From: Matthijs Douze Date: Mon, 15 May 2023 06:55:00 -0700 Subject: [PATCH] Update INSTALL.md (#2853) Summary: Update the INSTALL to get a recipe "that works" to install GPU faiss (which is not obvious). Pull Request resolved: https://github.com/facebookresearch/faiss/pull/2853 Reviewed By: algoriddle Differential Revision: D45652590 Pulled By: mdouze fbshipit-source-id: fededbdfda33e7a44d279e6bf16da316cff6fcb0 --- INSTALL.md | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/INSTALL.md b/INSTALL.md index c82f0dd507..021e3c81f7 100644 --- a/INSTALL.md +++ b/INSTALL.md @@ -31,13 +31,12 @@ $ conda install -c pytorch/label/nightly faiss-cpu $ conda install -c pytorch/label/nightly faiss-gpu ``` -A combination of versions that works with Pytorch (as of 2022-11-23): +A combination of versions that installs GPU Faiss with CUDA 11.4 and Pytorch (as of 2023-05-08): ``` -conda create -n faiss_1.7.3 python=3.8 -conda activate faiss_1.7.3 -conda install pytorch==1.11.0 cudatoolkit=11.3 -c pytorch -conda install numpy -conda install -c pytorch faiss-gpu=1.7.3 cudatoolkit=11.3 +conda create --name faiss_1.7.4 python=3.10 +conda activate faiss_1.7.4 +conda install faiss-gpu=1.7.4 -c pytorch -c nvidia +conda install faiss-gpu pytorch pytorch-cuda -c pytorch -c nvidia conda install -c conda-forge notebook conda install -y matplotlib ```